解决方式:修改Pytorch源码,我的目录在/home/efy/mambaforge/envs/py38/lib/python3.8/site-packages/torch/nn/modules/upsampling.py,大概是153行的位置。
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)