Smal updates

2024-06-30 22:18:16 +01:00 · 2024-06-30 22:18:16 +01:00 · e98cfa31fa
commit e98cfa31fa
parent 48227feba5
4 changed files with 163 additions and 18 deletions
--- a/src/training/histogram_dataset.py
+++ b/src/training/histogram_dataset.py
@ -18,8 +18,8 @@ class HistogramDataset(Dataset):
        self,
        /,
        paths: List[Path],
-        bin_count: int,
-        edit_count: int = 5,
+        bin_count: int = 16,
+        edit_count: int = 12,
        target_size=(240, 240),
        delete_corrupt_images: bool = False,
        cache_path: Optional[Path] = None,