logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

解决:RuntimeError: adaptive_max_pool2d_backward_cuda does not have........(添加注意力机制CBAM)

在train.py文件中,大概324行左右,修改下列代码。重新运行 train.py 文件。

文章图片
#服务器#目标检测
YOLO-v5训练自己的数据集

例如:打开terminal是,环境显示是base,而你需要进入pytorch环境,则输入以下命令,即可进入pytorch环境中。打开terminal,进入本项目的annoconda环境。运行train.py文件,即可开始训练,训练结果会保存在runs/train文件夹中。这里默认大家已经安装好自己的环境了并配置好了Pycharm。如果环境不是本项目所需的环境,需要激活,进行切换环境。所需要的包安装

文章图片
#深度学习
到底了