pytorch
diff --git a/‎.circleci/config.yml‎
Lines changed: 68 additions & 68 deletions b/‎.circleci/config.yml‎
Lines changed: 68 additions & 68 deletions
diff --git a/‎.circleci/regenerate.py‎
Lines changed: 1 addition & 1 deletion b/‎.circleci/regenerate.py‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎test/builtin_dataset_mocks.py‎
Lines changed: 38 additions & 0 deletions b/‎test/builtin_dataset_mocks.py‎
Lines changed: 38 additions & 0 deletions
@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
         for os_type in ["linux", "macos", "win"]:
             python_versions = PYTHON_VERSIONS
             cu_versions_dict = {
-                "linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],
+                "linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.5.2", "rocm5.0"],
                 "win": ["cpu", "cu113", "cu115"],
                 "macos": ["cpu"],
             }
 
@@ -911,6 +911,44 @@ def country211(info, root, config):
     return num_examples * len(classes)
 
 
+@register_mock
+def food101(info, root, config):
+    data_folder = root / "food-101"
+
+    num_images_per_class = 3
+    image_folder = data_folder / "images"
+    categories = ["apple_pie", "baby_back_ribs", "waffles"]
+    image_ids = []
+    for category in categories:
+        image_files = create_image_folder(
+            image_folder,
+            category,
+            file_name_fn=lambda idx: f"{idx:04d}.jpg",
+            num_examples=num_images_per_class,
+        )
+        image_ids.extend(path.relative_to(path.parents[1]).with_suffix("").as_posix() for path in image_files)
+
+    meta_folder = data_folder / "meta"
+    meta_folder.mkdir()
+
+    with open(meta_folder / "classes.txt", "w") as file:
+        for category in categories:
+            file.write(f"{category}\n")
+
+    splits = ["train", "test"]
+    num_samples_map = {}
+    for offset, split in enumerate(splits):
+        image_ids_in_split = image_ids[offset :: len(splits)]
+        num_samples_map[split] = len(image_ids_in_split)
+        with open(meta_folder / f"{split}.txt", "w") as file:
+            for image_id in image_ids_in_split:
+                file.write(f"{image_id}\n")
+
+    make_tar(root, f"{data_folder.name}.tar.gz", compression="gz")
+
+    return num_samples_map[config.split]
+
+
 @register_mock
 def dtd(info, root, config):
     data_folder = root / "dtd"
Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,`
`32`	`32`	`for os_type in ["linux", "macos", "win"]:`
`33`	`33`	`python_versions = PYTHON_VERSIONS`
`34`	`34`	`cu_versions_dict = {`
`35`		`- "linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.3.1", "rocm4.5.2"],`
	`35`	`+ "linux": ["cpu", "cu102", "cu113", "cu115", "rocm4.5.2", "rocm5.0"],`
`36`	`36`	`"win": ["cpu", "cu113", "cu115"],`
`37`	`37`	`"macos": ["cpu"],`
`38`	`38`	`}`