diff options
Diffstat (limited to 'nerv/examples/lmptb/rnn/tnn.lua')
| -rw-r--r-- | nerv/examples/lmptb/rnn/tnn.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/examples/lmptb/rnn/tnn.lua b/nerv/examples/lmptb/rnn/tnn.lua index ae9ed7a..8c3963c 100644 --- a/nerv/examples/lmptb/rnn/tnn.lua +++ b/nerv/examples/lmptb/rnn/tnn.lua @@ -198,7 +198,7 @@ function TNN:init(batch_size, chunk_size) end end -- initialize sub layers - ref.layer:init(batch_size) + ref.layer:init(batch_size, chunk_size) end local flags_now = {} |