diff options
| author | Qi Liu <[email protected]> | 2016-03-11 13:32:00 +0800 |
|---|---|---|
| committer | Qi Liu <[email protected]> | 2016-03-11 13:32:00 +0800 |
| commit | f26288ba61d3d16866e1b227a71e7d9c46923436 (patch) | |
| tree | ea41bb08994d9d2ee59c3ac5f3ec2c41bcaac6d2 /lua/main.lua | |
| parent | 05fcde5bf0caa1ceb70fef02fc88eda6f00c5ed5 (diff) | |
update mini_batch_init
Diffstat (limited to 'lua/main.lua')
| -rw-r--r-- | lua/main.lua | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/lua/main.lua b/lua/main.lua index 684efac..ce0270a 100644 --- a/lua/main.lua +++ b/lua/main.lua @@ -9,7 +9,7 @@ local timer = global_conf.timer timer:tic('IO') local data_path = 'nerv/nerv/examples/lmptb/PTBdata/' -local train_reader = nerv.Reader(data_path .. 'vocab', data_path .. 'ptb.train.txt.adds') +local train_reader = nerv.Reader(data_path .. 'vocab', data_path .. 'ptb.valid.txt.adds') local val_reader = nerv.Reader(data_path .. 'vocab', data_path .. 'ptb.valid.txt.adds') local train_data = train_reader:get_all_batch(global_conf) @@ -41,3 +41,5 @@ for i = 1, global_conf.max_iter do end timer:toc('global') timer:check('global') +timer:check('network') +timer:check('gc') |