diff --git a/use-cases/3dgan/dataloader.py b/use-cases/3dgan/dataloader.py index 65585779..5ac9c4b7 100644 --- 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 index 942efeb7..676424aa 100644 --- 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