Retraining on a trained costum weight

[icklerly1]

Hi,

I am planning on re-training using a costume weight, e.g. of epoch 5.
When I execute a training a with that as a weight-file it starts counting the train epochs at 5 instead of 0.
Is there a way to reset the epoch within the weight-file? A pretrained coco-weight starts at 0 but it certainly has trained more epochs :)

Thanks in advance!
Lydia

[LSH9832]

just modify the last line in your config file params/train/train_XXX.yaml as follows

force_start_epoch: 0             # set -1 to disable this option