search

victoresque / pytorch-template

PyTorch deep learning projects made easy.
MIT License
4.7k stars 1.08k forks source link

Save best model #81

Closed deeperlearner closed 3 years ago

deeperlearner commented 3 years ago

https://github.com/victoresque/pytorch-template/blob/a39a102914feace71ddd05b58d5e5a7c0e2c99dd/base/base_trainer.py#L101

here should correct as if epoch % self.save_period == 0 or best: so that it can save best model in the case self.save_period > 1.

deeperlearner commented 3 years ago

solution for #72