Recreate an average checkpoint

Hi,

During my last Locomotive training, I stopped the training and deleted a saved checkpoint.

Is there a way to recalculate and generate a new « average checkpoint » from a 32000 pt and a 35000 pt to continue the training or do I have to restart the training from scratch?

Something such : python train.py --config config.json --rebuild-average_model

Thank you.

Yes, you can call the average_models function via a Python script: Locomotive/onmt_tools.py at main · LibreTranslate/Locomotive · GitHub

2 Likes

Thank you @pierotofy !

1 Like