Extend CheckpointSaver class

facebookresearch / EGG

EGG: Emergence of lanGuage in Games

MIT License

281 stars 99 forks source link

Closed nicofirst1 closed 3 years ago

nicofirst1 commented 3 years ago

Extended the CheckpointSaver to add the following:

possibility to define the starting epoch. This comes in handy when you do not want to use the Trainer.load_checkpoint.
Define a max checkpoint files to keep at the same time. For big enough model and many epochs the checkpoint directory may become saturated with too many heavy files. With this functionality a max number of files are allowed in the directory at the same time. Oldest checkpoints are deleted first.

nicofirst1 commented 3 years ago

I cannot replicate the error on my machine. @robertodessi can you take a look?

robertodessi commented 3 years ago

Test don't fail now. Except for the comments it looks good to me, thanks!

robertodessi commented 3 years ago