pth 模型转 onnx时出错
参考
up3 = F.interpolate(output3, size=[output2.size(2), output2.size(3)], mode="nearest") # error
up3 = F.interpolate(output3, size=[32, 32], mode="nearest") # ok
经调查是上采样时的问题,将原来的变量换为固定数值解决
pth 模型转 onnx时出错
参考
up3 = F.interpolate(output3, size=[output2.size(2), output2.size(3)], mode="nearest") # error
up3 = F.interpolate(output3, size=[32, 32], mode="nearest") # ok
经调查是上采样时的问题,将原来的变量换为固定数值解决