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
dgl
Commits
1b2658b9
Unverified
Commit
1b2658b9
authored
Sep 14, 2020
by
Chao Ma
Committed by
GitHub
Sep 14, 2020
Browse files
update (#2192)
parent
1aa25cb6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
8 deletions
+0
-8
examples/pytorch/graphsage/experimental/train_dist.py
examples/pytorch/graphsage/experimental/train_dist.py
+0
-8
No files found.
examples/pytorch/graphsage/experimental/train_dist.py
View file @
1b2658b9
...
...
@@ -228,14 +228,6 @@ def run(args, device, data):
forward_time
+=
forward_end
-
start
backward_time
+=
compute_end
-
forward_end
# Aggregate gradients in multiple nodes.
if
not
args
.
standalone
:
for
param
in
model
.
parameters
():
if
param
.
requires_grad
and
param
.
grad
is
not
None
:
th
.
distributed
.
all_reduce
(
param
.
grad
.
data
,
op
=
th
.
distributed
.
ReduceOp
.
SUM
)
param
.
grad
.
data
/=
dgl
.
distributed
.
get_num_client
()
optimizer
.
step
()
update_time
+=
time
.
time
()
-
compute_end
...
...
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