diff options
Diffstat (limited to 'nerv/examples/asr_trainer.lua')
| -rw-r--r-- | nerv/examples/asr_trainer.lua | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/nerv/examples/asr_trainer.lua b/nerv/examples/asr_trainer.lua index 684ea30..5001e12 100644 --- a/nerv/examples/asr_trainer.lua +++ b/nerv/examples/asr_trainer.lua @@ -1,3 +1,5 @@ +require 'lfs' +require 'pl' local function build_trainer(ifname) local param_repo = nerv.ParamRepo() param_repo:import(ifname, nil, gconf) @@ -133,6 +135,8 @@ local trainer_defaults = { local options = make_options(trainer_defaults) table.insert(options, {"help", "h", "boolean", default = false, desc = "show this help information"}) +table.insert(options, {"dir", nil, "string", + default = nil, desc = "specify the working directory"}) arg, opts = nerv.parse_args(arg, options) @@ -156,8 +160,19 @@ check_and_add_defaults(trainer_defaults) local pf0 = gconf.initialized_param local trainer = build_trainer(pf0) local accu_best = trainer(nil, gconf.cv_scp, false) +local date_pattern = "%Y%m%d%H%M%S" +local logfile_name = "log" +local working_dir = opts["dir"].val or string.format("nerv_%s", os.date(date_pattern)) print_gconf() +if not lfs.mkdir(working_dir) then + nerv.error("[asr_trainer] working directory already exists") +end +-- copy the network config +dir.copyfile(arg[1], working_dir) +-- set logfile path +nerv.set_logfile(path.join(working_dir, logfile_name)) +path.chdir(working_dir) nerv.info("initial cross validation: %.3f", accu_best) for i = 1, gconf.max_iter do nerv.info("[NN] begin iteration %d with lrate = %.6f", i, gconf.lrate) @@ -168,7 +183,7 @@ for i = 1, gconf.max_iter do string.gsub( (string.gsub(pf0[1], "(.*/)(.*)", "%2")), "(.*)%..*", "%1"), - os.date("%Y%m%d%H%M%S"), + os.date(date_pattern), i, gconf.lrate, accu_tr), gconf.cv_scp, false) |