Add an option to automatically extract and save weights when saving checkpoints #71
Description
Just a suggestion that i think would help a lot, to add an option that extracts and saves a weight file automatically when saving a checkpoint, ./weights/$experimentName_$steps.pth, to easily watch and read progress