Tensor: return F.interpolate(input, self.size, self.scal_upsampl">
赞
踩
提示
File "/home/taobotics/rosenv/lib/python3.7/site-packages/torch/nn/modules/upsampling.py", line 154
去到这个目录修改upsampling.py
下的
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)
改为
def forward(self, input: Tensor) -> Tensor:
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/SSS__jq/article/details/123458804
————————————————
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。