From 20beee0c5b0396bdc1b6601d82e9499e22f91ca0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 21 Mar 2019 22:51:22 +0200 Subject: [PATCH] updates --- train.py | 2 +- utils/gcp.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 12638c0a..c130089c 100644 --- a/train.py +++ b/train.py @@ -205,7 +205,7 @@ if __name__ == '__main__': parser.add_argument('--multi-scale', action='store_true', help='random image sizes per batch 320 - 608') parser.add_argument('--img-size', type=int, default=32 * 13, help='pixels') parser.add_argument('--resume', action='store_true', help='resume training flag') - parser.add_argument('--num_workers', type=int, default=0, help='number of Pytorch DataLoader workers') + parser.add_argument('--num-workers', type=int, default=0, help='number of Pytorch DataLoader workers') opt = parser.parse_args() print(opt, end='\n\n') diff --git a/utils/gcp.sh b/utils/gcp.sh index 6d46f236..03ac0111 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -9,7 +9,7 @@ sudo shutdown # Start sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 cp -r weights yolov3 -cd yolov3 && python3 train.py --batch-size 64 --multi_scale +cd yolov3 && python3 train.py --batch-size 16 --num-workers 4 # Resume python3 train.py --resume