yolov5执行测试代码出现报错

我执行的dectect.py出现了如下报错

我的版本是
torch                          1.8.1
torchvision                    0.4.2+cu100
python                       3.6.3
请各位大神帮帮看下~

torchvision  版本过低了，而且torch1.8建议python版本是高于3.8的，你先版本弄好看看。而且官方的文件好像没有这个gen_wts.py文件的吧