Create data/data_config.py

data/data_config.py

@@ -0,0 +1,14 @@
+from dataclasses import dataclass, field
+from typing import Optional, List
+
+
+@dataclass
+class SingleDataConfig:
+    dataset_folder: str
+    imagefolder: bool = True
+    dataset_weight: float = 1.0 # Not used yet
+
+@dataclass
+class DataConfig:
+    datasets: List[SingleDataConfig]
+    val_dataset: Optional[SingleDataConfig] = None