Fix train/test split

2024-06-04 22:48:07 +01:00 · 2024-06-04 22:48:07 +01:00 · edeac12e37
commit edeac12e37
parent af56ec3fec
7 changed files with 17239 additions and 33632 deletions
--- a/src/editor/training/histogram_dataset.py
+++ b/src/editor/training/histogram_dataset.py
@ -24,6 +24,8 @@ class HistogramDataset(Dataset):
        cache_path: Optional[Path] = None,
    ):
        self._paths = sorted(paths)
+        logging.info(f"Loaded {len(self._paths)} original images")
+
        self._edit_count = edit_count
        self._bin_count = bin_count
        self._target_size = target_size