How do I train the model based on a pre_trained model?

Support
1

I’m new to this software.Now I want to train a model based on a pre_trained model, how do I make some changes in file ‘.yaml’ to achieve it or how do I make some changes in the code files.

2

Hello!

Use -train_from in the command, for example:

onmt_train -config config.yml -train_from model/model_step_n.pt

https://opennmt.net/OpenNMT-py/options/train.html?highlight=train_from#Initialization

In the yaml file, you don’t have to change anything (unless you want to). Just make sure that train_steps value is enough to cover the number of steps of both the original training and the new training.

Kind regards,
Yasmin

2 Likes