new build model

2026-10-10 02:23:29 +02:00
parent 787e2f47ec
commit 00ad59a59b
11 changed files with 255 additions and 24 deletions
@@ -2,9 +2,10 @@


 for ((c = 1; c <= 32; c *= 2)); do
-    python -m src.main yolo.train --train-dir ./data/faces_scenery/train/ --val-dir ./data/faces_scenery/test/ \
-        --no-augment-val --c-hidden "$c" --batch-size 64 --epochs 200 --run-name "bn-weighted-loss-c$c" \
-        --no-save --train-workers 4 --image-size 64 --grid 6
+    python -m src.main yolo.train --train-dir ./data/wider_faces_coco/train/ --val-dir ./data/wider_faces_coco/test/ \
+        --no-augment-val --c-hidden "$c" --batch-size 64 --epochs 200 --run-name "bn-wider-faces-c$c" \
+        --no-save --train-workers 4 --image-size 256 --grid 8 \
+        --learning-rate 1e-3 --weight-decay 0.05 --warmup-epochs 3 --min-learning-rate 1e-5
 done