facebookresearch / FFCV-SSL

FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.
Apache License 2.0
203 stars 14 forks source link

fix checkpointing offset #3

Closed GregoireMialon closed 1 year ago

GregoireMialon commented 1 year ago

The restarting job will otherwise restart at epoch epoch with the weights obtained at the end of epoch in the stopped job, hence silently adding one epoch to the training