diff options
| author | txh18 <cloudygooseg@gmail.com> | 2015-11-08 17:59:27 +0800 |
|---|---|---|
| committer | txh18 <cloudygooseg@gmail.com> | 2015-11-08 17:59:27 +0800 |
| commit | a8d369d49933ffdd306f47db6b74e0d27deda5d0 (patch) | |
| tree | 8a2126153fb2363ffea94bfe8c0b9e88f11bd0ea /nerv/examples/lmptb/rnn/tnn.lua | |
| parent | abc36052969ab121c8a1cfa478fc14e9e8dc78a2 (diff) | |
trying to test softmax_ce_t
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 = {} |