From b7b9f7a7fc3a2e6a027175e5a84541ca2291fbb5 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 1 Jan 2023 11:36:00 +0100 Subject: Fixed accuracy calc, other improvements --- 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 803271b..af36d9e 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.1) + valid_set_size = self.valid_set_size if self.valid_set_size is not None else int(num_images * 0.2) valid_set_size = max(valid_set_size, 1) train_set_size = num_images - valid_set_size -- cgit v1.2.3-70-g09d2