AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘

Table of Contents

背景

运行python语言实现的深度学习模型【选用pytorch】代码时,代码报错

AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

方案

将报错文件“D:\Anaconda3\envs\yolov5-6.0\lib\site-packages\torch\nn\modules\upsampling.py”【具体路径请查看报错信息】大体类似

里边的forward方法修改一下,改成以下

def forward(self, input: Tensor) -> Tensor:
    return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)

修改后程序运行正常

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年9月21日
下一篇 2023年9月21日

相关推荐

此站出售,如需请站内私信或者邮箱!