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
0081afb8
Commit
0081afb8
authored
May 25, 2018
by
Christian Sarofeen
Browse files
Decrease default message size.
parent
cae6005c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
apex/parallel/distributed.py
apex/parallel/distributed.py
+2
-2
No files found.
apex/parallel/distributed.py
View file @
0081afb8
...
...
@@ -47,13 +47,13 @@ class DistributedDataParallel(Module):
Args:
module: Network definition to be run in multi-gpu/distributed mode.
message_size (Default = 10
0
e6): Minimum number of elements in a communication bucket.
message_size (Default = 10e6): Minimum number of elements in a communication bucket.
shared_param (Default = False): If your model uses shared parameters this must be true,
it will disable bucketing of parameters which is necessary to avoid race conditions.
"""
def
__init__
(
self
,
module
,
message_size
=
10000000
0
,
shared_param
=
False
):
def
__init__
(
self
,
module
,
message_size
=
10000000
,
shared_param
=
False
):
super
(
DistributedDataParallel
,
self
).
__init__
()
self
.
warn_on_half
=
True
if
dist
.
_backend
==
dist
.
dist_backend
.
GLOO
else
False
self
.
shared_param
=
shared_param
...
...
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