added single node distdataparallel

2026-09-10 12:21:57 +08:00 · 2019-07-03 15:24:56 -04:00
parent 6eb7674e18
commit 5f57792131
1 changed files with 1 additions and 0 deletions
@@ -293,6 +293,7 @@ class Trainer(TrainerIO):
        """ This is called before pickling. """
        state = self.__dict__.copy()
        del state['cluster']
+        del state['experiment']
        return state

    def __dp_train(self, gpu_nb, proc_rank):