diff options
author | Qi Liu <liuq901@163.com> | 2016-05-12 17:41:21 +0800 |
---|---|---|
committer | Qi Liu <liuq901@163.com> | 2016-05-12 17:41:21 +0800 |
commit | c0fdb7ee2966546023410bb03e62dee0cf64e0e1 (patch) | |
tree | cd90688b8aee2991a78f971e1bc4e0e9596e357b /nerv/examples/ptb/main.lua | |
parent | d88a57f4852c50a2678de950ee650ed9b6a895f0 (diff) | |
parent | 10916f721a945a5edd052ab93027413fd3c01f65 (diff) |
Merge branch 'master' into 'master'
beta-1.2
multiple change
1. Merge affine & lstm_gate & projection layer
2. Change clip behavior
3. Seq_buffer support sequence level shuffle
4. LSTM & LSTMP layer support multiple input
See merge request !4
Diffstat (limited to 'nerv/examples/ptb/main.lua')
-rw-r--r-- | nerv/examples/ptb/main.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/examples/ptb/main.lua b/nerv/examples/ptb/main.lua index 5d1a326..7191da0 100644 --- a/nerv/examples/ptb/main.lua +++ b/nerv/examples/ptb/main.lua @@ -105,7 +105,7 @@ function trainer:epoch_preprocess(dataset, do_train) total_frame = 0 end -function trainer:mini_batch_middleprocess(cnt, info) +function trainer:mini_batch_inprocess(cnt, info) for t = 1, gconf.chunk_size do local tmp = info.output[1][t]:new_to_host() for i = 1, gconf.batch_size do |