diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index de596da41b756..286fa292878d0 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -84,7 +84,7 @@ def forward(self, x): """ if self.trainer.proc_rank == 0: - print(*args, **kwargs) + log.info(*args, **kwargs) @abstractmethod def forward(self, *args, **kwargs):