From b01e6aa2b27f49aeab04b448f09e0ef9e689ea53 Mon Sep 17 00:00:00 2001 From: David Hou Date: Fri, 3 May 2024 00:06:53 -0700 Subject: [PATCH] drop cookie before data_get? --- examples/mlperf/model_train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/mlperf/model_train.py b/examples/mlperf/model_train.py index 8e34242fb0..b261ceed20 100644 --- a/examples/mlperf/model_train.py +++ b/examples/mlperf/model_train.py @@ -195,11 +195,11 @@ def train_resnet(): pt = time.perf_counter() + prev_cookie = None # free previous cookie after gpu work has been enqueued try: next_proc = data_get(it) except StopIteration: next_proc = None - prev_cookie = None # free previous cookie after gpu work has been enqueued dt = time.perf_counter() @@ -258,11 +258,11 @@ def train_resnet(): (loss, top_1), y, proc = eval_step(proc[0], proc[1]), proc[2], proc[3] # drop inputs, keep cookie + prev_cookie = None # free previous cookie after gpu work has been enqueued try: next_proc = data_get(it) except StopIteration: next_proc = None - prev_cookie = None # free previous cookie after gpu work has been enqueued loss, top_1 = loss.numpy().item(), top_1.numpy().item() num_samples = sum(yi != -1 for yi in y)