diff options
Diffstat (limited to 'data')
| -rw-r--r-- | data/csv.py | 2 |
1 files changed, 1 insertions, 1 deletions
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(): | |||
| 151 | 151 | ||
| 152 | num_images = len(items) | 152 | num_images = len(items) |
| 153 | 153 | ||
| 154 | valid_set_size = self.valid_set_size if self.valid_set_size is not None else int(num_images * 0.1) | 154 | valid_set_size = self.valid_set_size if self.valid_set_size is not None else int(num_images * 0.2) |
| 155 | valid_set_size = max(valid_set_size, 1) | 155 | valid_set_size = max(valid_set_size, 1) |
| 156 | train_set_size = num_images - valid_set_size | 156 | train_set_size = num_images - valid_set_size |
| 157 | 157 | ||