diff options
| author | txh18 <[email protected]> | 2015-12-03 00:01:24 +0800 |
|---|---|---|
| committer | txh18 <[email protected]> | 2015-12-03 00:01:24 +0800 |
| commit | 63cd5b0ab0d2fd1693fdaec0e57b5e02ad718dfb (patch) | |
| tree | b303832e6e0f42d22e39c8ab1e3028410414e47f | |
| parent | b1ccbf0cc57dcbca443724377278c88bf2d707a7 (diff) | |
...
| -rw-r--r-- | nerv/examples/lmptb/tnn/tnn.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/nerv/examples/lmptb/tnn/tnn.lua b/nerv/examples/lmptb/tnn/tnn.lua index 8cafe6a..56c9dc0 100644 --- a/nerv/examples/lmptb/tnn/tnn.lua +++ b/nerv/examples/lmptb/tnn/tnn.lua @@ -386,6 +386,7 @@ function TNN:propagate_dfs(ref, t) --ok, do propagate --print("debug ok, propagating"); + --The MB moving will cause bordering history to be changed, so it is more wise to flush the input activation if (bit.band(self.feeds_now.flagsPack_now[t], bit.bor(nerv.TNN.FC.SEQ_START, nerv.TNN.FC.SEQ_END)) > 0) then --flush cross-border history for i = 1, self.batch_size do local seq_start = bit.band(self.feeds_now.flags_now[t][i], nerv.TNN.FC.SEQ_START) |