search

ku21fan / STR-Fewer-Labels

Scene Text Recognition (STR) methods trained with fewer real labels (CVPR 2021)
MIT License
173 stars 27 forks source link

Train from checkpoint #11

Closed lyminhuit closed 1 year ago

lyminhuit commented 1 year ago

Hi, I train model on colab, after a while it will be disconnected. I can train again after a few hours, but I have to train again from the author's checkpoint. How can I continue to train from the checkpoint (weight) that I trained last time.

Thanks!

ku21fan commented 1 year ago

Hi,

When running train.py, use this option --saved_model saved_models/{the folder including your checkpoint}/best_score.pth. Then your training continues from your last checkpoint.

Hope it helps :)