temporarily disable cache for imagneet

priyakasimbeg · priyakasimbeg · commit 56337f8e209a · 2026-01-15T23:22:24.000Z
diff --git a/algoperf/workloads/imagenet_resnet/imagenet_pytorch/workload.py b/algoperf/workloads/imagenet_resnet/imagenet_pytorch/workload.py
@@ -220,10 +220,10 @@ def _build_dataset(
       )
 
     folder = 'train' if 'train' in split else 'val'
-    dataset = CachedImageFolder(
+    dataset = ImageFolder(
       os.path.join(data_dir, folder),
       transform=transform_config,
-      cache_file='.imagenet_cache_index.json',
+      # cache_file='.imagenet_cache_index.json',
     )
 
     if split == 'eval_train':

Original file line number	Diff line number	Diff line change
`@@ -220,10 +220,10 @@ def _build_dataset(`
`220`	`220`	`)`
`221`	`221`
`222`	`222`	`folder = 'train' if 'train' in split else 'val'`
`223`		`- dataset = CachedImageFolder(`
	`223`	`+ dataset = ImageFolder(`
`224`	`224`	`os.path.join(data_dir, folder),`
`225`	`225`	`transform=transform_config,`
`226`		`- cache_file='.imagenet_cache_index.json',`
	`226`	`+ # cache_file='.imagenet_cache_index.json',`
`227`	`227`	`)`
`228`	`228`
`229`	`229`	`if split == 'eval_train':`