debug

2019-08-07 08:03:40 -04:00 · 2019-08-07 08:03:40 -04:00 · 0527a1dad1
parent 2018380598
commit 0527a1dad1
1 changed files with 1 additions and 1 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -86,7 +86,7 @@ def test_cpu_restore_training():

    # set the epoch start hook so we can predict before the model does the full training
    def assert_pred_same():
-        assert trainer.current_epoch == real_global_epoch and trainer.real_global_epoch > 0
+        assert trainer.current_epoch == real_global_epoch and trainer.current_epoch > 0

        # predict with loaded model to make sure answers are the same
        trainer.model.eval()