torch-rnn icon indicating copy to clipboard operation
torch-rnn copied to clipboard

Add epoch and validation loss to checkpoint

Open calittle opened this issue 7 years ago • 1 comments

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. From Karpathy's char-rnn implementation.

calittle avatar Apr 14 '17 12:04 calittle

Looks good. I think versioning checkpoints on more parameters is a sensible idea. Quick question - why do we need to store both the epoch value, and iteration number? Can epochs not be derived from iteration count?

ChrisCummins avatar Jul 31 '17 10:07 ChrisCummins