最近更新于 2022-06-02 20:37

我遇到这个错误的地方:PyTorch 1.11.0 下,在 YOLOv5 v6.1 使用官方提供的预训练模型测试时报错。

根据报错(上面我选中的部分),编辑这个文件,找到指定的错误行,上面指出的 154 行

将 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)

作者 IYATT-yx