当前位置:   article > 正文

YOLOv5报错(Upsample)_assertion `upsample11' failed.

assertion `upsample11' failed.

错误提示

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

点击这个错误提示到

修改如下:

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

  • 1
  • 2
  • 3
  • 4
  • 5

在这里插入图片描述
注释掉,增加一行,删除不兼容的参数即可。

出现这个问题的原因应该是安装的pytorch版本不兼容导致。
这里的方法是实用性解决。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/码创造者/article/detail/888897
推荐阅读
相关标签
  

闽ICP备14008679号