yolov5报错：Upsample‘ object has no attribute ‘recompute_scale_factor

最新推荐文章于 2024-04-23 23:52:35 发布

Yukee_

最新推荐文章于 2024-04-23 23:52:35 发布

阅读量98

点赞数

文章标签： YOLO 深度学习人工智能 pytorch

本文链接：https://blog.csdn.net/Yukee_/article/details/132298957

版权

修改pytorch包中的源码
路径：D:\Anaconda3\envs\pytorch\Lib\site-packages\torch\nn\modules
修改如下：注释掉的是原来的代码

 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)