Why --resume does not work

TimDettmers / sparse_learning

Sparse learning library and sparse momentum resources.

MIT License

377 stars 45 forks source link

Why --resume does not work #23

Open AlexiaAnn opened 2 years ago

AlexiaAnn commented 2 years ago

command: python mnist_cifar/main.py --model wrn-28-2 --data cifar --resume models/model.pt --start-epoch 20 The model is still retrained and cannot be trained from where it was interrupted。why?

TimDettmers commented 2 years ago

I believe this is an issue with printing the wrong epoch number. From the code, it seems it should work right, but the logged epoch number starts again from 1.