diff --git a/utils/gcp.sh b/utils/gcp.sh index 2ffe8e01..4bb0776b 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -116,7 +116,7 @@ sudo shutdown #Docker sudo docker kill $(sudo docker ps -q) sudo docker pull ultralytics/yolov3:v0 -sudo nvidia-docker run -it --ipc=host --mount type=bind,source="$(pwd)"/coco,target=/usr/src/coco ultralytics/yolov3:v0 +sudo nvidia-docker run -it --ipc=host --mount type=bind,source="$(pwd)"/coco,target=/usr/src/coco ultralytics/yolov3:v1 clear while true @@ -124,7 +124,7 @@ do python3 train.py --weights '' --prebias --img-size 512 --batch-size 32 --accumulate 2 --evolve --epochs 27 --bucket yolov4/512_coco_27e --device 0 done -python3 train.py --data data/coco.data --img-size 320 --batch-size 64 --accumulate 1 --epochs 1 --adam --device 1 --prebias +python3 train.py --weights '' --prebias --img-size 512 --batch-size 8 --accumulate 8 --epochs 27 --device 0 while true; do python3 train.py --data data/coco.data --img-size 320 --batch-size 64 --accumulate 1 --evolve --epochs 1 --adam --bucket yolov4/adamdefaultpw_coco_1e; done