Unverified Commit 0ced22c2 authored by Qidong Su's avatar Qidong Su Committed by GitHub
Browse files

fix (#2142)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-10-127.us-west-2.compute.internal>
Co-authored-by: default avatarChao Ma <mctt90@gmail.com>
parent 67e520c4
...@@ -233,7 +233,7 @@ def run(args, device, data): ...@@ -233,7 +233,7 @@ def run(args, device, data):
step_t = time.time() - tic_step step_t = time.time() - tic_step
step_time.append(step_t) step_time.append(step_t)
iter_tput.append(num_seeds / (step_t)) iter_tput.append(len(blocks[-1].dstdata[dgl.NID]) / step_t)
if step % args.log_every == 0: if step % args.log_every == 0:
acc = compute_acc(batch_pred, batch_labels) acc = compute_acc(batch_pred, batch_labels)
gpu_mem_alloc = th.cuda.max_memory_allocated() / 1000000 if th.cuda.is_available() else 0 gpu_mem_alloc = th.cuda.max_memory_allocated() / 1000000 if th.cuda.is_available() else 0
......
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