From a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Wed, 4 Jan 2023 22:06:05 +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 c505230..a60733a 100644 --- a/data/csv.py +++ b/data/csv.py @@ -151,7 +151,7 @@ class CSVDataModule(): num_images = len(items) - valid_set_size = self.valid_set_size if self.valid_set_size is not None else int(num_images * 0.2) + valid_set_size = self.valid_set_size if self.valid_set_size is not None else num_images // 10 valid_set_size = max(valid_set_size, 1) train_set_size = num_images - valid_set_size -- cgit v1.2.3-70-g09d2