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

fix resnet eval (#4507)

Browse Source
This commit is contained in:
chenyu
2024-05-10 00:49:00 -04:00
committed by GitHub
parent a602dc67d3
commit b399d98e41
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/mlperf/model_train.py
View File

@@ -268,10 +268,11 @@ def train_resnet():
BEAM.value = EVAL_BEAM
if INITMLPERF:
i, proc = 0, fake_data_get(EVAL_BS)
else:
it = iter(tqdm(batch_load_resnet(batch_size=EVAL_BS, val=True, shuffle=False, pad_first_batch=True), total=steps_in_val_epoch))
i, proc = 0, data_get(it)
else:
i, proc = 0, fake_data_get(EVAL_BS)
prev_cookies = []
while proc is not None:
GlobalCounters.reset()