Toggle navigation
首页
(current)
问答
文章
话题
商城
登录
注册
5
pytorch利用onnx模块导出xxx.proto文件的时候出错
```python import torch def main(): data=torch.rand(1, 2, 2, 3) print("input", data) print(data.size()) model=torch.nn.Upsample((4, 6), mode='bilinear', align_corners=True) pre=model(data) print("output", pre) print(pre.size()) print("Transform the model to proto") torch.onnx.export(model, data, "model.proto", verbose=True) main() ``` 出现以下问题 ```bash TypeError: upsample_bilinear2d() got an unexpected keyword argument 'align_corners' (occurred when translating upsample_bilinear2d) ``` 我的pytorch版本是0.4.0 我的onnx版本是1.3.0
0 条评论
请先
登录
后评论
默认排序
时间排序
0 个回答
您需要登录后才可以回答问题,
登录
或者
注册
关注
1
关注
收藏
0
收藏,
439
浏览
无需付费
LeonSun
提出于 2018-09-04 17:31
相似问题
×
发送私信
发给:
内容: