Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

updated args

This commit is contained in:
William Falcon 2019-06-25 18:59:37 -04:00
parent bf3b86ce4d
commit 0fd4d5e7a1
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch_lightning/models/trainer.py
View File

@ -14,6 +14,7 @@ try:
except ModuleNotFoundError: except ModuleNotFoundError:
APEX_AVAILABLE = False APEX_AVAILABLE = False
class Trainer(TrainerIO): class Trainer(TrainerIO):
def __init__(self, def __init__(self,
@ -66,8 +67,6 @@ class Trainer(TrainerIO):
self.data_parallel_device_ids = gpus self.data_parallel_device_ids = gpus
self.data_parallel = gpus is not None and len(gpus) > 0 self.data_parallel = gpus is not None and len(gpus) > 0
pdb.set_trace()
# training state # training state
self.optimizers = None self.optimizers = None
self.prog_bar = None self.prog_bar = None