(已解决)RuntimeError: “slow_conv2d_cpu“ not implemented for ‘Half‘

具体错误如下:

 File "E:\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\conv.py", line 460, in forward
    return self._conv_forward(input, self.weight, self.bias)
  File "E:\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\conv.py", line 456, in _conv_forward
    return F.conv2d(input, weight, bias, self.stride,
RuntimeError: "slow_conv2d_cpu" not implemented for 'Half'

如果按照精度问题修改还是会报错

我的解决方法是看看配置文件的参数和模块是否都完整成功的导入

比如yolo.py文件或者general.py文件:

from utils.general import LOGGER, check_version,  check_yaml,colorstr, make_divisible,print_args
from utils.torch_utils import (
    fuse_conv_and_bn,
    initialize_weights,
    model_info,
    profile,
    scale_img,
    select_device,
    time_sync,
)

补齐配置文件的参数就迎刃而解啦~

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值