diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index de596da41b..286fa29287 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -84,7 +84,7 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks): """ if self.trainer.proc_rank == 0: - print(*args, **kwargs) + log.info(*args, **kwargs) @abstractmethod def forward(self, *args, **kwargs):