将Pytorch模型转为ONNX 时出现shape 为负数情况,该问题还是由于ONNX不支持一些Pytorch的函数。
torch.repeat_interleave();
解决方法:
将repeat_interleave函数替换为interpolate(近邻插值法)
使用torch_nn_func.interpolate()替换torch.repeat_interleave();其中mode = 'nearest'</
将Pytorch模型转为ONNX 时出现shape 为负数情况,该问题还是由于ONNX不支持一些Pytorch的函数。
torch.repeat_interleave();
解决方法:
将repeat_interleave函数替换为interpolate(近邻插值法)
使用torch_nn_func.interpolate()替换torch.repeat_interleave();其中mode = 'nearest'</