use log no print (#940)

This commit is contained in:
William Falcon
2020-02-25 13:06:48 -05:00
committed by GitHub
parent 5dd2afeab1
commit 5d89fed2a6
+1 -1
View File
@@ -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):