Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
apex
Commits
12b49b98
Unverified
Commit
12b49b98
authored
Jul 18, 2018
by
ngimel
Committed by
GitHub
Jul 18, 2018
Browse files
delete __getstate__ arg
parent
77ee4bcd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
apex/parallel/distributed.py
apex/parallel/distributed.py
+1
-1
No files found.
apex/parallel/distributed.py
View file @
12b49b98
...
...
@@ -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'
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment