diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-01 11:36:00 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-01 11:36:00 +0100 |
| commit | b7b9f7a7fc3a2e6a027175e5a84541ca2291fbb5 (patch) | |
| tree | 24fd6d9f3a92ce9f5cccd5cdd914edfee665af71 /data | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-b7b9f7a7fc3a2e6a027175e5a84541ca2291fbb5.tar.gz textual-inversion-diff-b7b9f7a7fc3a2e6a027175e5a84541ca2291fbb5.tar.bz2 textual-inversion-diff-b7b9f7a7fc3a2e6a027175e5a84541ca2291fbb5.zip | |
Fixed accuracy calc, other improvements
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 | ||