Add option to save model for each epochs

[camillebrianceau]

It would be great to add an option to save the models from all other epochs, regardless of performance and not only the best-loss.