added single node distdataparallel
This commit is contained in:
parent
251c2e964f
commit
5797f812ad
|
@ -292,7 +292,6 @@ class Trainer(TrainerIO):
|
||||||
def __getstate__(self):
|
def __getstate__(self):
|
||||||
""" This is called before pickling. """
|
""" This is called before pickling. """
|
||||||
state = self.__dict__.copy()
|
state = self.__dict__.copy()
|
||||||
del state['namelist']
|
|
||||||
return {}
|
return {}
|
||||||
|
|
||||||
def __dp_train(self, gpu_nb, proc_rank):
|
def __dp_train(self, gpu_nb, proc_rank):
|
||||||
|
|
Loading…
Reference in New Issue