Move training/evaluation data to unified directory

RobinDong · May 10, 2024 · 225c70d · 225c70d
1 parent 64ba0c7
commit 225c70d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tinymm/CLIP/evaluate.py b/tinymm/CLIP/evaluate.py
@@ -10,8 +10,8 @@
 
 
 class Imagenet1K:
-    image_path: str = "/home/robin/Downloads/imagenet/val"
-    map_file_path: str = "/home/robin/Downloads/imagenet_map.txt"
+    image_path: str = "/data/imagenet/val"
+    map_file_path: str = "/data/imagenet/imagenet_map.txt"
     nr_categories: int = 1000
     seq_len: int = 64
     image_size: tuple = (256, 256)

diff --git a/tinymm/model_config.py b/tinymm/model_config.py
@@ -10,10 +10,10 @@ class ModelConfig:
 @dataclass
 class TrainConfig:
     data_path: tuple = (
-        "/home/robin/Downloads/cc12m",
-        "/data/coco_captions/",
-        "/home/robin/Downloads/cc3m",
-        "/home/robin/Downloads/sbu_caption",
+        "/data/cc12m",
+        "/data/coco_captions",
+        "/data/cc3m",
+        "/data/sbu_caption",
     )
     eval_ratio: float = 0.05
     num_workers: int = 4