use cache

priyakasimbeg · priyakasimbeg · commit 4dc8e868acd3 · 2026-01-29T22:09:00.000Z
diff --git a/algoperf/workloads/imagenet_resnet/imagenet_pytorch/workload.py b/algoperf/workloads/imagenet_resnet/imagenet_pytorch/workload.py
@@ -220,7 +220,7 @@ def _build_dataset(
       )
 
     folder = 'train' if 'train' in split else 'val'
-    dataset = ImageFolder(
+    dataset = CachedImageFolder(
       os.path.join(data_dir, folder),
       transform=transform_config,
       cache_file='.imagenet_{}_cache_index.json'.format(split),

Original file line number	Diff line number	Diff line change
`@@ -220,7 +220,7 @@ def _build_dataset(`
`220`	`220`	`)`
`221`	`221`
`222`	`222`	`folder = 'train' if 'train' in split else 'val'`
`223`		`- dataset = ImageFolder(`
	`223`	`+ dataset = CachedImageFolder(`
`224`	`224`	`os.path.join(data_dir, folder),`
`225`	`225`	`transform=transform_config,`
`226`	`226`	`cache_file='.imagenet_{}_cache_index.json'.format(split),`