From e0b686b475885f0c8480f7173eaa7359adf17e27 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 26 Dec 2022 14:24:21 +0100 Subject: Set default dimensions to 768; add config inheritance --- data/csv.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'data') 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(): prompt_processor: PromptProcessor, class_subdir: str = "cls", num_class_images: int = 1, - size: int = 512, + size: int = 768, repeats: int = 1, dropout: float = 0, interpolation: str = "bicubic", @@ -196,7 +196,7 @@ class CSVDataset(Dataset): prompt_processor: PromptProcessor, batch_size: int = 1, num_class_images: int = 0, - size: int = 512, + size: int = 768, repeats: int = 1, dropout: float = 0, interpolation: str = "bicubic", -- cgit v1.2.3-54-g00ecf