diff options
Diffstat (limited to 'data/csv.py')
| -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 0810c2c..0ad36dc 100644 --- a/data/csv.py +++ b/data/csv.py | |||
| @@ -51,7 +51,7 @@ class CSVDataModule(): | |||
| 51 | prompt_processor: PromptProcessor, | 51 | prompt_processor: PromptProcessor, |
| 52 | class_subdir: str = "cls", | 52 | class_subdir: str = "cls", |
| 53 | num_class_images: int = 1, | 53 | num_class_images: int = 1, |
| 54 | size: int = 512, | 54 | size: int = 768, |
| 55 | repeats: int = 1, | 55 | repeats: int = 1, |
| 56 | dropout: float = 0, | 56 | dropout: float = 0, |
| 57 | interpolation: str = "bicubic", | 57 | interpolation: str = "bicubic", |
| @@ -196,7 +196,7 @@ class CSVDataset(Dataset): | |||
| 196 | prompt_processor: PromptProcessor, | 196 | prompt_processor: PromptProcessor, |
| 197 | batch_size: int = 1, | 197 | batch_size: int = 1, |
| 198 | num_class_images: int = 0, | 198 | num_class_images: int = 0, |
| 199 | size: int = 512, | 199 | size: int = 768, |
| 200 | repeats: int = 1, | 200 | repeats: int = 1, |
| 201 | dropout: float = 0, | 201 | dropout: float = 0, |
| 202 | interpolation: str = "bicubic", | 202 | interpolation: str = "bicubic", |