Accuracy on test set is {} based on the best model {} ?????????

[lemo2012]

In you codes, the accuracy on test set is based on the latest model after all the train step finished, not based on the best model saved in checkpoint. Am I right?

[changukshin]

No. Please see step 6.2 in train.py

The variable ‘path’ is the path of best model.
Not the latest model.

We assume the least validation loss model is the best model.

Please look forward about ‘validation based early stopping’.