model is not loading from last epoch and iteration

howardyclo / pytorch-seq2seq-example

Fully batched seq2seq example based on practical-pytorch, and more extra features.

76 stars 17 forks source link

model is not loading from last epoch and iteration #5

Open ghost opened 5 years ago

ghost commented 5 years ago

Hello,

I am learning about seq2seq model, I am trying to run your model, It ran perfect but when I am trying to load from last saved state it picks up perfectly from last global step but it does not start from the last iteration and epoch i.e. It starts again from 0 iterations and 0 epoch. Can you please suggest me a possible solution for this?.

howardyclo commented 5 years ago

@sid972 Hi, I think you can also save iteration/epoch step in checkpoint, then reload them if you wanna retrain the model.