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