ADD: skip download option

interTwin-eu · Nov 9, 2023 · be0c115 · be0c115
1 parent 3bcc410
commit be0c115
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/use-cases/3dgan/dataloader.py b/use-cases/3dgan/dataloader.py
@@ -25,6 +25,10 @@ def __init__(
         self.data_url = data_url
 
     def load(self):
+        if self.data_path is None:
+            print("Data path is None. Skipping dataset downloading")
+            return
+
         # Download data
         if not os.path.exists(self.data_path):
             gdown.download_folder(

diff --git a/use-cases/3dgan/pipeline.yaml b/use-cases/3dgan/pipeline.yaml
@@ -4,7 +4,7 @@ executor:
     steps:
       - class_path: dataloader.Lightning3DGANDownloader
         init_args:
-          data_path: exp_data/
+          data_path: exp_data/ # Set to null to skip dataset download
           data_url: https://drive.google.com/drive/folders/1uPpz0tquokepptIfJenTzGpiENfo2xRX
 
       - class_path: trainer.Lightning3DGANTrainer