added single node distdataparallel

This commit is contained in:
William Falcon
2019-07-03 16:31:43 -04:00
parent 080c308bcc
commit 7010d16752
-6
View File
@@ -285,12 +285,6 @@ class Trainer(TrainerIO):
else:
self.__run_pretrain_routine(model)
def __getstate__(self):
""" This is called before pickling. """
state = self.__dict__.copy()
# del state['experiment']
return state
def dp_train(self, gpu_nb, proc_rank, model):
"""
Entry point into a DP thread