mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
added single node distdataparallel
This commit is contained in:
1 parent
080c308bcc
commit
7010d16752
1 file changed
-6
@@ -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
|
||||
|
||||
Reference in new issue
Block a user