summaryrefslogtreecommitdiffstats
path: root/data/csv.py
diff options
context:
space:
mode:
Diffstat (limited to 'data/csv.py')
-rw-r--r--data/csv.py4
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 ):