"git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "efd909e62e1099c36441dd9cba565828e3b7becf"
Unverified Commit 79b057f0 authored by nv-dlasalle's avatar nv-dlasalle Committed by GitHub
Browse files

Fix example to test_acc is always computed at last epoch (#3014)

parent 2570d412
...@@ -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,
......
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