现业务需要基于ms实现对torch版本的复现,并将torch训练的模型权重导入ms,发现upsample算子在输入一致的情况下输出不同,请问ms中如何实现torch版本的upsample操作?
****************************************************解答*****************************************************
你可以设置ResizeBilinear的参数,nn.ResizeBilinear(half_pixel_centers=True),结果应该就一致了。
可以暂时把报错(raise ValueError(f"Currently `half_pixel_centers`=True only support in Ascend device_target, ")对应的校验device_target代码删掉就可以了,在nn_ops.py文件里的 ResizeBilinear类的__init__方法里