From 5d89fed2a69a32df9d988218de2010aa1eae0eca Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Feb 2020 13:06:48 -0500 Subject: [PATCH] use log no print (#940) --- pytorch_lightning/core/lightning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index de596da4..286fa292 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):