diff options
Diffstat (limited to 'nerv/examples/lmptb/m-tests/tnn_test.lua')
| -rw-r--r-- | nerv/examples/lmptb/m-tests/tnn_test.lua | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/nerv/examples/lmptb/m-tests/tnn_test.lua b/nerv/examples/lmptb/m-tests/tnn_test.lua index a2c38f0..c4890b6 100644 --- a/nerv/examples/lmptb/m-tests/tnn_test.lua +++ b/nerv/examples/lmptb/m-tests/tnn_test.lua @@ -238,15 +238,15 @@ valid_fn = data_dir .. '/ptb.valid.txt.adds' test_fn = data_dir .. '/ptb.test.txt.adds' global_conf = { - lrate = 0.1, wcost = 1e-6, momentum = 0, + lrate = 1, wcost = 1e-6, momentum = 0, cumat_type = nerv.CuMatrixFloat, mmat_type = nerv.MMatrixFloat, nn_act_default = 0, hidden_size = 200, - chunk_size = 15, - batch_size = 1, - max_iter = 25, + chunk_size = 5, + batch_size = 10, + max_iter = 20, param_random = function() return (math.random() / 5 - 0.1) end, train_fn = train_fn, |