From e2b0f0ead13be29476c047dfb49ad49f85a849bb Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Fri, 8 Nov 2024 14:33:59 -0800 Subject: [PATCH] fix Signed-off-by: Hemil Desai --- nemo/lightning/pytorch/trainer.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/nemo/lightning/pytorch/trainer.py b/nemo/lightning/pytorch/trainer.py index 0cba14076c52..01f9a57de313 100644 --- a/nemo/lightning/pytorch/trainer.py +++ b/nemo/lightning/pytorch/trainer.py @@ -49,15 +49,6 @@ def configure_no_restart_validation_training_loop(trainer: pl.Trainer) -> None: class Trainer(pl.Trainer, IOMixin): - def _configure_no_restart_validation_training_loop(self) -> None: - if not isinstance(self.fit_loop.epoch_loop, _TrainingEpochLoop): - warnings.warn("Detected custom epoch loop. Skipping no validation on restart support.", UserWarning) - return - - ## Pass trainer object to avoid trainer getting overwritten as None - loop = NoValOnRestartTrainingLoop(self, self.min_steps, self.max_steps) - self.fit_loop.epoch_loop = loop - def add_io(self, obj): """Recurse to the leaves of a container and add io functionality to non-serializable leaves""" if isinstance(obj, (dict, list)):