Unverified Commit 12b49b98 authored by ngimel's avatar ngimel Committed by GitHub
Browse files

delete __getstate__ arg

parent 77ee4bcd
......@@ -92,7 +92,7 @@ class DistributedDataParallel(Module):
super(DistributedDataParallel, self).__setstate__(state)
self.reduction_stream = torch.cuda.Stream()
def __getstate__(self, state):
def __getstate__(self):
attrs = copy.copy(self.__dict__)
if dist._backend != dist.dist_backend.NCCL:
del attrs['self.reduction_stream']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment