BCE to CE lconf + batch size 16

2018-10-11 17:43:34 +02:00
parent d336e0053d
commit 24a41972cb
2 changed files with 3 additions and 5 deletions
@@ -26,7 +26,6 @@ cd yolov3/checkpoints
 wget https://storage.googleapis.com/ultralytics/yolov3.pt
 cp yolov3.pt latest.pt
 cd ..
-python3 train.py -img_size 416 -batch_size 12 -epochs 1 -resume 1
+python3 train.py -img_size 416 -batch_size 16 -epochs 1 -resume 1
 python3 test.py -img_size 416 -weights_path checkpoints/latest.pt -conf_thres 0.5

-