yolov5-resnet训练问题解决

1.首先安装torch、torchvision、progress、matplotlib

pip install torch -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple

pip install progress -i https://pypi.tuna.tsinghua.edu.cn/simple

2.将cifar.py下的240行的asyncTrue删除

3.出现下面爆红,torchvision版本不匹配,升级torchvision版本

pip install --upgrade torchvision

4.训练TRAINING.MD

5.如果没有安装cuda,就会报下面这个

点击链接,将cuda改为cpu

5.将view改为reshape,继续训练TRAINING.md

6.出现以下报错,将.data[0],[0]改为.item()

7.最后再运行TRAINING.md

选用自己想用的模型

 

相关推荐

  1. yolov5训练长条型目标召回率低问题

    2024-06-05 19:36:45       30 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-05 19:36:45       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-05 19:36:45       106 阅读
  3. 在Django里面运行非项目文件

    2024-06-05 19:36:45       87 阅读
  4. Python语言-面向对象

    2024-06-05 19:36:45       96 阅读

热门阅读

  1. Shell 编程之免交互

    2024-06-05 19:36:45       32 阅读
  2. SQL注入的危害和原理

    2024-06-05 19:36:45       32 阅读
  3. 达梦数据库的锁介绍

    2024-06-05 19:36:45       31 阅读
  4. Python中的None用法深度解析

    2024-06-05 19:36:45       35 阅读
  5. C++的算法:贪心算法

    2024-06-05 19:36:45       27 阅读
  6. mysql中的函数

    2024-06-05 19:36:45       30 阅读
  7. 前端-Vue组件key的作用

    2024-06-05 19:36:45       28 阅读
  8. Go 延迟调用机制

    2024-06-05 19:36:45       28 阅读
  9. Python实现连连看4

    2024-06-05 19:36:45       29 阅读