pytorch1.4 转onnx遇到的问题

分类: 文章 • 2024-06-29 16:14:34

pytorch 不同版本下载
https://download.pytorch.org/whl/torch_stable.html

F.interpolate(x, size=(x_copy.size(2), x_copy.size(3)),
mode=“bilinear”, align_corners=True) # 原始
pytorch1.4 转onnx遇到的问题

为了不改变原始的写法，需要将opset_version=9　改为11
pytorch1.4 转onnx遇到的问题
即可解决