InstantIR / data /data_config.py
fffiloni's picture
Upload folder using huggingface_hub
0305a63 verified
raw
history blame contribute delete
330 Bytes
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