lmjohns3 / theanets

Neural network toolkit for Python
http://theanets.rtfd.org
MIT License
328 stars 74 forks source link

feature request: checkpointing #44

Closed majidaldo closed 9 years ago

majidaldo commented 9 years ago

I'm seeing the argument for periodic saving on the command line. However, I'm not seeing its implementation in the code. I suggest a periodic saving based on some elapsed time (as opposed to training cycles for example).

Given the long training times, this should be prioritized.

lmjohns3 commented 9 years ago

Thanks for the poke!

Model saving is enabled by providing both

If model saving is enabled and there is a file with the correct name at the start of training, that model will be loaded before training. This could be an issue if people make multiple calls to .train() but I'm going to leave it alone for now.

I've only tried this out on the mnist examples, so if you notice anything wrong, please reopen this issue.