"examples/sampling/vscode:/vscode.git/clone" did not exist on "03406bd96d4d01b5a32378937a176cb434642519"
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):
vend = time.time()
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()
if (queue is not None) or (proc_id == 0):
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