diff --git a/train.py b/train.py index a83fa1c0..20efba44 100644 --- a/train.py +++ b/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 diff --git a/utils/evolve.sh b/utils/evolve.sh new file mode 100644 index 00000000..34e3d223 --- /dev/null +++ b/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 \ No newline at end of file