updated args

This commit is contained in:
William Falcon
2019-06-25 18:59:37 -04:00
parent bf3b86ce4d
commit 0fd4d5e7a1
+1 -2
View File
@@ -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