mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

drop cookie before data_get?

Browse Source
This commit is contained in:
David Hou
2024-05-03 00:06:53 -07:00
parent 3c42d960af
commit b01e6aa2b2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/mlperf/model_train.py
View File

@@ -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)