updated args

2019-06-25 18:59:37 -04:00 · 2019-06-25 18:59:37 -04:00 · 0fd4d5e7a1
parent bf3b86ce4d
commit 0fd4d5e7a1
1 changed files with 1 additions and 2 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -14,6 +14,7 @@ try:
 except ModuleNotFoundError:
    APEX_AVAILABLE = False

+
 class Trainer(TrainerIO):

    def __init__(self,
@ -66,8 +67,6 @@ class Trainer(TrainerIO):
        self.data_parallel_device_ids = gpus
        self.data_parallel = gpus is not None and len(gpus) > 0

-        pdb.set_trace()
-
        # training state
        self.optimizers = None
        self.prog_bar = None