From 2ad46871e2ead985445da2848a4eb7072b6e48aa Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 14 Nov 2022 17:09:58 +0100 Subject: Update --- data/csv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'data') diff --git a/data/csv.py b/data/csv.py index 793fbf8..67ac43b 100644 --- a/data/csv.py +++ b/data/csv.py @@ -93,7 +93,7 @@ class CSVDataModule(pl.LightningDataModule): items = [item for item in items if not "skip" in item or item["skip"] != True] num_images = len(items) - valid_set_size = int(num_images * 0.2) + valid_set_size = int(num_images * 0.1) if self.valid_set_size: valid_set_size = min(valid_set_size, self.valid_set_size) valid_set_size = max(valid_set_size, 1) -- cgit v1.2.3-54-g00ecf