报错信息: 

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

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐