diff options
Diffstat (limited to 'nerv/examples/asr_trainer.lua')
-rw-r--r-- | nerv/examples/asr_trainer.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nerv/examples/asr_trainer.lua b/nerv/examples/asr_trainer.lua index aa1019d..38ba6e9 100644 --- a/nerv/examples/asr_trainer.lua +++ b/nerv/examples/asr_trainer.lua @@ -196,6 +196,7 @@ local trainer_defaults = { max_iter = 20, min_halving = 5, do_halving = false, + keep_halving = false, cumat_tname = "nerv.CuMatrixFloat", mmat_tname = "nerv.MMatrixFloat", debug = false, @@ -294,6 +295,8 @@ for i = gconf.cur_iter, gconf.max_iter do if gconf.accu_best - accu_prev < gconf.start_halving_inc and i >= gconf.min_halving then gconf.do_halving = true + elseif not gconf.keep_halving then + gconf.do_halving = false end if gconf.do_halving then gconf.lrate = gconf.lrate * gconf.halving_factor |