updates

train.py

@@ -446,7 +446,7 @@ if __name__ == '__main__':
         mixed_precision = False
 
     # scale hyp['obj'] by img_size (evolved at 416)
-    hyp['obj'] *= opt.img_size / 416.
+    hyp['obj'] *= opt.img_size / 320.
 
     tb_writer = None
     if not opt.evolve:  # Train normally

utils/evolve.sh

@@ -0,0 +1,10 @@
+#for i in 1 2 3 4 5 6 7
+#do
+#  t=ultralytics/yolov3:v139 && sudo docker pull $t && sudo nvidia-docker run --d --ipc=host -v "$(pwd)"/coco:/usr/src/coco $t utils/evolve.sh $i
+#  sleep 30
+# done
+
+while true
+do
+  python3 train.py --data coco2014.data --img-size 320 --epochs 27 --batch-size 64 --accumulate 1 --evolve --weights '' --pre --bucket yolov4/320_coco2014_27e --device $1
+done