diff --git a/examples/mlperf/dataloader.py b/examples/mlperf/dataloader.py index b513ccbc08..0942e83615 100644 --- a/examples/mlperf/dataloader.py +++ b/examples/mlperf/dataloader.py @@ -129,7 +129,7 @@ def batch_load_resnet(batch_size=64, val=False, shuffle=True, seed=None, pad_fir q_in, q_out = Queue(), Queue() sz = (batch_size*BATCH_COUNT, 224, 224, 3) - shm_name = "resnet_X_val" if val else "resnet_X_train", + shm_name = "resnet_X_val" if val else "resnet_X_train" if not OSX and os.path.exists(f"/dev/shm/{shm_name}"): os.unlink(f"/dev/shm/{shm_name}") shm = shared_memory.SharedMemory(name=shm_name, create=True, size=prod(sz)) procs = []