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

fix resnet eval (#4507)

Browse source
This commit is contained in:
chenyu 2024-05-10 00:49:00 -04:00 • committed by GitHub
commit b399d98e41
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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()