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 | ): |
