From 17e4926cdf06d601ebcbb4dcff906387765a58cd Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 2 Jun 2020 23:40:14 -0700 Subject: [PATCH] updates --- Dockerfile | 3 ++- models/yolov5x.yaml | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index c506a3e86..9a8cd6a9a 100644 --- a/Dockerfile +++ b/Dockerfile @@ -40,7 +40,8 @@ COPY . /usr/src/app # sudo docker run --gpus all --ipc=host ultralytics/yolov5:latest while true; do python3 train.py --evolve; done # Bash into running container -# sudo docker container exec -it 97919ad657de /bin/bash +# sudo docker container exec -it ba65811811ab bash +# python -c "from utils.utils import *; create_backbone('weights/last.pt')" && gsutil cp weights/backbone.pt gs://* # Bash into stopped container # sudo docker commit 6d525e299258 user/test_image && sudo docker run -it --gpus all --ipc=host -v "$(pwd)"/coco:/usr/src/coco --entrypoint=sh user/test_image diff --git a/models/yolov5x.yaml b/models/yolov5x.yaml index 3c01f1813..1efc980e9 100644 --- a/models/yolov5x.yaml +++ b/models/yolov5x.yaml @@ -26,7 +26,7 @@ backbone: # yolov5 head head: - [[-1, 3, Bottleneck, [1024]], # 11 + [[-1, 3, Bottleneck, [1024, False]], # 11 [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large) [-2, 1, nn.Upsample, [None, 2, 'nearest']],