报错信息:
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'
原因:
Upsample对象没有recompute_scale_factor属性
解决:
找到upsampling.py文件的forward函数,如下:
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)
将原本的forward函数注释了(别的项目使用可能还要改回去),改为:
def forward(self, input: Tensor) -> Tensor:
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)