search

tensorflow / nmt

TensorFlow Neural Machine Translation Tutorial
Apache License 2.0
6.39k stars 1.96k forks source link

How to train the model again with larger training steps? #151

Closed ghost closed 7 years ago

ghost commented 7 years ago

I've trained an NMT system with (say) 50,000 training steps initially. Now, I would like to train it for (say) 75000 steps. If I change the num_train_steps=75000 and re-run the training code, it is just loading the old hparams and not training further.

In other implementations like nmt.matlab and nmt.hybrid, I used to do the similar thing and it worked.

What am I missing here?

oahziur commented 7 years ago

@ssokhey In additional to your original training command, adding --override_loaded_hparams=True. I don't think this feature is available at tf-1.2 branch though.

ghost commented 7 years ago

I didn't notice override tag. Thanks @oahziur