From 0466285f59b2b73dbb0187f7883df6b577e032ab Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 16 Nov 2019 22:09:15 -0800 Subject: [PATCH] updates --- utils/gcp.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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