Glenn Jocher
4 years ago
2 changed files with 3 additions and 2 deletions
+ 2
- 1
Dockerfile
View File
| # sudo docker run --gpus all --ipc=host ultralytics/yolov5:latest while true; do python3 train.py --evolve; done | # sudo docker run --gpus all --ipc=host ultralytics/yolov5:latest while true; do python3 train.py --evolve; done | ||||
| # Bash into running container | # 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 | # 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 | # 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 |
+ 1
- 1
models/yolov5x.yaml
View File
| # yolov5 head | # yolov5 head | ||||
| 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) | [-1, 1, nn.Conv2d, [na * (nc + 5), 1, 1, 0]], # 12 (P5/32-large) | ||||
| [-2, 1, nn.Upsample, [None, 2, 'nearest']], | [-2, 1, nn.Upsample, [None, 2, 'nearest']], |
Loading…