error on test.py

[jerrywgz]

I try to test yolov5l on v100 using followed commend

python test.py --img-size 640 --conf-thres 0.1 --batch-size 16 --weights weights/yolov5l.pt --data data/coco.yaml --device 0

RuntimeError is reported, but it works on detect.py and train.py

I used the docker image (ultralytics/yolov5:latest) you provided.

[github-actions]

Hello @jerrywgz, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Google Colab Notebook, Docker Image, and GCP Quickstart Guide for example environments.

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom model or data training question, please note that Ultralytics does not provide free personal support. As a leader in vision ML and AI, we do offer professional consulting, from simple expert advice up to delivery of fully customized, end-to-end production solutions for our clients, such as:

• Cloud-based AI surveillance systems operating on hundreds of HD video streams in realtime.
• Edge AI integrated into custom iOS and Android apps for realtime 30 FPS video inference.
• Custom data training, hyperparameter evolution, and model exportation to any destination.

For more information please visit https://www.ultralytics.com.

[glenn-jocher]

@jerrywgz thank you for the bug report! This looks familiar, I believe it's caused by a thop dependency used for calculating FLOPs. You can pip uninstall thop to fix this in your container.

We've pushed a fixed image to dockerhub as well, so you if you pull you should receive the fix now.