feat(ml):rename dataloader

jolibrain · Oct 1, 2024 · f760b1e · f760b1e
1 parent 98b1315
commit f760b1e
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/data/self_supervised_vid_mask_online_dataset.py b/data/self_supervised_vid_mask_online_dataset.py
@@ -18,7 +18,7 @@ def natural_keys(text):
     return [atoi(c) for c in re.split("(\d+)", text)]
 
 
-class SelfSupervisedTemporalLabeledMaskOnlineDataset(BaseDataset):
+class SelfSupervisedVidMaskOnlineDataset(BaseDataset):
     def __len__(self):
         """Return the total number of images in the dataset.
         As we have two datasets with potentially different number of images,

diff --git a/examples/example_ddpm_vid_mario.json b/examples/example_ddpm_vid_mario.json
@@ -117,7 +117,7 @@
     },
     "data": {
         "crop_size": 64,
-        "dataset_mode": "self_supervised_temporal_labeled_mask_online",
+        "dataset_mode": "self_supervised_vid_mask_online",
         "direction": "AtoB",
         "image_bits": 8,
         "inverted_mask": false,

diff --git a/options/common_options.py b/options/common_options.py
@@ -664,7 +664,7 @@ def initialize(self, parser):
                 "aligned",
                 "nuplet_unaligned_labeled_mask",
                 "temporal_labeled_mask_online",
-                "self_supervised_temporal_labeled_mask_online",
+                "self_supervised_vid_mask_online",
                 "self_supervised_temporal",
                 "single",
                 "unaligned_labeled_mask_ref",

diff --git a/tests/test_run_vid_diffusion_online.py b/tests/test_run_vid_diffusion_online.py
@@ -54,7 +54,7 @@
 
 models_diffusion = ["palette"]
 G_netG = ["unet_vid"]
-data_dataset_mode = ["self_supervised_temporal_labeled_mask_online"]
+data_dataset_mode = ["self_supervised_vid_mask_online"]
 
 product_list = product(models_diffusion, G_netG, data_dataset_mode)