最近更新于 2024-05-05 14:18
我遇到这个错误的地方：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)
AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’