marvin/CNN_Website
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

what ever

Browse Source
This commit is contained in:
marvin
2026-10-18 14:47:15 +02:00
parent eaa6e2f303
commit 9b75386acf
5 changed files with 41 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/train_test.sh
+2 -2
View File
@@ -2,8 +2,8 @@
for ((c = 1; c <= 32; c *= 2)); do
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" \
python -m src.main yolo.train --train-dir ./data/faces_combined/train/ --val-dir ./data/faces_combined/test/ \
--no-augment-val --c-hidden "$c" --batch-size 64 --epochs 200 --run-name "bn-combined-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