diff options
Diffstat (limited to 'nerv/examples')
| -rw-r--r-- | nerv/examples/ptb/main.lua | 2 | ||||
| -rw-r--r-- | nerv/examples/trainer.lua | 1 |
2 files changed, 2 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 diff --git a/nerv/examples/trainer.lua b/nerv/examples/trainer.lua index f6c7a5a..caed2e2 100644 --- a/nerv/examples/trainer.lua +++ b/nerv/examples/trainer.lua @@ -80,6 +80,7 @@ local trainer_defaults = { local options = make_options(trainer_defaults) local extra_opt_spec = { + {"clip", nil, "number"}, {"resume-from", nil, "string"}, {"help", "h", "boolean", default = false, desc = "show this help information"}, {"dir", nil, "string", desc = "specify the working directory"}, |