diff options
Diffstat (limited to 'nerv/examples/lmptb/m-tests/dagl_test.lua')
-rw-r--r-- | nerv/examples/lmptb/m-tests/dagl_test.lua | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/nerv/examples/lmptb/m-tests/dagl_test.lua b/nerv/examples/lmptb/m-tests/dagl_test.lua index 9f45b6a..8959a04 100644 --- a/nerv/examples/lmptb/m-tests/dagl_test.lua +++ b/nerv/examples/lmptb/m-tests/dagl_test.lua @@ -140,8 +140,8 @@ global_conf = { mmat_type = nerv.CuMatrixFloat, hidden_size = 20, - batch_size = 5, - seq_size = 3, + chunk_size = 5, + batch_size = 3, max_iter = 18, param_random = function() return (math.random() / 5 - 0.1) end, independent = true, @@ -161,4 +161,4 @@ global_conf.vocab:build_file(global_conf.train_fn, false) local paramRepo = prepare_parameters(global_conf, true) local layerRepo = prepare_layers(global_conf, paramRepo) local dagL = prepare_dagLayer(global_conf, layerRepo) ---dagL:init(global_conf.batch_size) +dagL:init(global_conf.batch_size, global_conf.chunk_size) |