wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

change print to logging (#457)

Browse Source 
* change print to logging

* always use logging.info

* use f-strings

* update code style

* set logging configs

* remove unused code
This commit is contained in:
Ir1dXD
2019-11-05 08:43:21 -05:00
committed by William Falcon
parent 9a5307dc30
commit 5a9afb11cc
12 changed files with 56 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/root_module.py
+2 -1
View File
@@ -10,6 +10,7 @@ from pytorch_lightning.root_module.hooks import ModelHooks
from pytorch_lightning.root_module.memory import ModelSummary
from pytorch_lightning.root_module.model_saving import ModelIO
from pytorch_lightning.trainer.trainer_io import load_hparams_from_tags_csv
import logging
class LightningModule(GradInformation, ModelIO, ModelHooks):
@@ -240,7 +241,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks):
def summarize(self, mode):
model_summary = ModelSummary(self, mode=mode)
print(model_summary)
logging.info(model_summary)
def freeze(self):
for param in self.parameters():