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
79b057f0
"git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "efd909e62e1099c36441dd9cba565828e3b7becf"
Unverified
Commit
79b057f0
authored
Jun 12, 2021
by
nv-dlasalle
Committed by
GitHub
Jun 12, 2021
Browse files
Fix example to test_acc is always computed at last epoch (#3014)
parent
2570d412
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
examples/pytorch/rgcn/entity_classify_mp.py
examples/pytorch/rgcn/entity_classify_mp.py
+1
-1
No files found.
examples/pytorch/rgcn/entity_classify_mp.py
View file @
79b057f0
...
@@ -385,7 +385,7 @@ def run(proc_id, n_gpus, n_cpus, args, devices, dataset, split, queue=None):
...
@@ -385,7 +385,7 @@ def run(proc_id, n_gpus, n_cpus, args, devices, dataset, split, queue=None):
vend
=
time
.
time
()
vend
=
time
.
time
()
validation_time
+=
(
vend
-
vstart
)
validation_time
+=
(
vend
-
vstart
)
if
epoch
>
0
and
do_test
:
if
epoch
==
args
.
n_epochs
-
1
or
(
epoch
>
0
and
do_test
)
:
tstart
=
time
.
time
()
tstart
=
time
.
time
()
if
(
queue
is
not
None
)
or
(
proc_id
==
0
):
if
(
queue
is
not
None
)
or
(
proc_id
==
0
):
test_logits
,
test_seeds
=
evaluate
(
model
,
embed_layer
,
test_logits
,
test_seeds
=
evaluate
(
model
,
embed_layer
,
...
...
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