added single node distdataparallel

This commit is contained in:
William Falcon
2019-07-03 15:24:16 -04:00
parent 5ff3a90a6f
commit 6eb7674e18
+1 -1
View File
@@ -293,7 +293,7 @@ class Trainer(TrainerIO):
""" This is called before pickling. """
state = self.__dict__.copy()
del state['cluster']
return {}
return state
def __dp_train(self, gpu_nb, proc_rank):
"""