torch-rnn icon indicating copy to clipboard operation
torch-rnn copied to clipboard

How can I continue training from pretrained model?

Open erogol opened this issue 8 years ago • 2 comments

Does the library provide a function to continue the training from a pretrained model?

erogol avatar Jan 07 '17 11:01 erogol

Yes. Just use -init_from <path_to_checkpoint> train option.

https://github.com/jcjohnson/torch-rnn/blob/master/train.lua#L20

mlajtos avatar Feb 10 '17 19:02 mlajtos

That currently does not re-initialize all parameters, you need to look at what is saved and what is not. I am working on a new checkpointing system that will make resuming much easier, so stay tuned.

dgcrouse avatar Apr 27 '17 04:04 dgcrouse