use log no print (#940)

2020-02-25 13:06:48 -05:00 · 2020-02-25 13:06:48 -05:00 · 5d89fed2a6
parent 5dd2afeab1
commit 5d89fed2a6
1 changed files with 1 additions and 1 deletions
--- 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):