From a671e352fc48483221ffb55779aab6058efce414 Mon Sep 17 00:00:00 2001 From: calittle Date: Fri, 14 Apr 2017 08:33:08 -0400 Subject: [PATCH] Add epoch and validation loss to checkpoint Update the checkpoint filename to include the epoch value and validation loss value so we sample against a checkpoint with a lower (or not!) validation loss. --- train.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.lua b/train.lua index fa00af16..53d3c0bd 100644 --- a/train.lua +++ b/train.lua @@ -227,7 +227,7 @@ for i = start_i + 1, num_iterations do memory_usage = memory_usage, i = i } - local filename = string.format('%s_%d.json', opt.checkpoint_name, i) + local filename = string.format('%s_%d_epoch%.2f_%.4f.json', opt.checkpoint_name, i, epoch, val_loss) -- Make sure the output directory exists before we try to write it paths.mkdir(paths.dirname(filename)) utils.write_json(filename, checkpoint) @@ -237,7 +237,7 @@ for i = start_i + 1, num_iterations do model:clearState() model:float() checkpoint.model = model - local filename = string.format('%s_%d.t7', opt.checkpoint_name, i) + local filename = string.format('%s_%d_epoch%.2f_$.4f.t7', opt.checkpoint_name, i, epoch, val_loss) paths.mkdir(paths.dirname(filename)) torch.save(filename, checkpoint) model:type(dtype)