diff options
author | txh18 <cloudygooseg@gmail.com> | 2015-12-06 19:18:07 +0800 |
---|---|---|
committer | txh18 <cloudygooseg@gmail.com> | 2015-12-06 19:18:07 +0800 |
commit | 4e3cf34a29afbdb17c42ab7f1efacec52efd0e8b (patch) | |
tree | 9da189094ac9964ba4566802e1bb496e13ca64ab /nerv/examples | |
parent | 313011f24dfdacfe9c18d018d5bb877625a09ec7 (diff) |
added extend_t for tnn to save GPU memory
Diffstat (limited to 'nerv/examples')
-rw-r--r-- | nerv/examples/lmptb/bilstmlm_ptb_main.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/examples/lmptb/bilstmlm_ptb_main.lua b/nerv/examples/lmptb/bilstmlm_ptb_main.lua index cf0009b..2aba1d2 100644 --- a/nerv/examples/lmptb/bilstmlm_ptb_main.lua +++ b/nerv/examples/lmptb/bilstmlm_ptb_main.lua @@ -364,7 +364,7 @@ else nerv.printf("%s no user setting, all default...\n", global_conf.sche_log_pre) end -global_conf.work_dir = global_conf.work_dir_base .. 'h' .. global_conf.hidden_size .. 'l' .. global_conf.layer_num .. 'ch' .. global_conf.chunk_size .. 'ba' .. global_conf.batch_size .. 'slr' .. global_conf.lrate .. 'wc' .. global_conf.wcost .. 'dr' .. global_conf.dropout_str +global_conf.work_dir = global_conf.work_dir_base .. 'h' .. global_conf.hidden_size .. 'l' .. global_conf.layer_num --.. 'ch' .. global_conf.chunk_size .. 'ba' .. global_conf.batch_size .. 'slr' .. global_conf.lrate .. 'wc' .. global_conf.wcost .. 'dr' .. global_conf.dropout_str global_conf.train_fn_shuf = global_conf.work_dir .. '/train_fn_shuf' global_conf.train_fn_shuf_bak = global_conf.train_fn_shuf .. '_bak' global_conf.param_fn = global_conf.work_dir .. "/params" |