mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

drop cookie before data_get?

Browse source
This commit is contained in:
David Hou 2024-05-03 00:06:53 -07:00
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)