diff options
Diffstat (limited to 'data')
-rw-r--r-- | data/csv.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/data/csv.py b/data/csv.py index c5e7aef..81e8b6b 100644 --- a/data/csv.py +++ b/data/csv.py | |||
@@ -187,7 +187,7 @@ class VlpnDataModule(): | |||
187 | dropout: float = 0, | 187 | dropout: float = 0, |
188 | shuffle: bool = False, | 188 | shuffle: bool = False, |
189 | interpolation: str = "bicubic", | 189 | interpolation: str = "bicubic", |
190 | color_jitter: bool = True, | 190 | color_jitter: bool = False, |
191 | template_key: str = "template", | 191 | template_key: str = "template", |
192 | placeholder_tokens: list[str] = [], | 192 | placeholder_tokens: list[str] = [], |
193 | valid_set_size: Optional[int] = None, | 193 | valid_set_size: Optional[int] = None, |
@@ -372,7 +372,7 @@ class VlpnDataset(IterableDataset): | |||
372 | dropout: float = 0, | 372 | dropout: float = 0, |
373 | shuffle: bool = False, | 373 | shuffle: bool = False, |
374 | interpolation: str = "bicubic", | 374 | interpolation: str = "bicubic", |
375 | color_jitter: bool = True, | 375 | color_jitter: bool = False, |
376 | generator: Optional[torch.Generator] = None, | 376 | generator: Optional[torch.Generator] = None, |
377 | npgenerator: Optional[np.random.Generator] = None, | 377 | npgenerator: Optional[np.random.Generator] = None, |
378 | ): | 378 | ): |