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