Skip to content

Commit 9ad4200

Browse files
committed
Remove expected on_train_epoch_start when restarting mid epoch
1 parent c59ab40 commit 9ad4200

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tests/tests_pytorch/models/test_hooks.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -660,8 +660,6 @@ def test_trainer_model_hook_system_fit_no_val_and_resume_max_steps(tmp_path):
660660
{"name": "train_dataloader"},
661661
{"name": "Callback.on_train_start", "args": (trainer, model)},
662662
{"name": "on_train_start"},
663-
{"name": "Callback.on_train_epoch_start", "args": (trainer, model)},
664-
{"name": "on_train_epoch_start"},
665663
*model._train_batch(trainer, model, steps_after_reload, trainer.strategy.root_device, current_batch=1),
666664
{"name": "Callback.on_train_epoch_end", "args": (trainer, model)},
667665
{"name": "on_train_epoch_end"}, # before ModelCheckpoint because it's a "monitoring callback"

0 commit comments

Comments
 (0)