debug

Author: awaelchli

pytorch_lightning/loops/epoch/evaluation_epoch_loop.py

@@ -100,6 +100,9 @@ def advance(
         if batch is None:
             raise StopIteration
 
+        assert self.num_dataloaders is not None
+        self.trainer.logger_connector.on_evaluation_batch_start(batch, batch_idx, dataloader_idx, self.num_dataloaders)
+
         with self.trainer.profiler.profile("evaluation_batch_to_device"):
             batch = self.trainer.accelerator.batch_to_device(batch, dataloader_idx=dataloader_idx)
 
@@ -172,9 +175,6 @@ def on_evaluation_batch_start(self, batch: Any, batch_idx: int, dataloader_idx:
         """
         self.trainer.logger_connector.on_batch_start()
 
-        assert self.num_dataloaders is not None
-        self.trainer.logger_connector.on_evaluation_batch_start(batch, batch_idx, dataloader_idx, self.num_dataloaders)
-
         if self.trainer.testing:
             self.trainer.call_hook("on_test_batch_start", batch, batch_idx, dataloader_idx)
         else: