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
pl_examples/basic_examples/lightning_module_template.py
+4 -3
View File
@@ -2,6 +2,7 @@
Example template for defining a system
"""
import os
import logging
from argparse import ArgumentParser
from collections import OrderedDict
@@ -214,17 +215,17 @@ class LightningTemplateModel(LightningModule):
@pl.data_loader
def train_dataloader(self):
print('training data loader called')
logging.info('training data loader called')
return self.__dataloader(train=True)
@pl.data_loader
def val_dataloader(self):
print('val data loader called')
logging.info('val data loader called')
return self.__dataloader(train=False)
@pl.data_loader
def test_dataloader(self):
print('test data loader called')
logging.info('test data loader called')
return self.__dataloader(train=False)
@staticmethod