search

NikolasEnt / soccernet-calibration-sportlight

SoccerNet@CVPR | 1st place solution for Camera Calibration Challenge 2023
27 stars 3 forks source link

Restarting training from checkpoint #8

Closed petem24 closed 1 month ago

petem24 commented 1 month ago

Firstly thank you for making this repo public!

I am currently training the keypoint model and i got to epoch 13. I am now restarting from the checkpoint pth file by adding the path in train_config.yaml but it starts again from epoch -1, am I starting again from the beginning or does this not matter?

Apologies if this is obvious but this is my first time training a model

NikolasEnt commented 1 month ago

Thank you for your interest in the project.

If you set the model checkpoint path in model.params.pretrain, the model is loaded and you are continuing training from the weights saved in the checkpoint file. But epoch number in the logs will start from 0, also the new epoch numbering will be used for LR scheduler.