首页 > 其他 > 详细

问题记录贴

时间:2020-07-14 16:21:28      阅读:164      评论:0      收藏:0      [点我收藏+]

torch转onnx

在转一个并不复杂的模型的时候出现错误.模型并不存在什么复杂的算子.

RuntimeError: tuple appears in op that does not forward tuples (VisitNode at /pytorch/torch/csrc/jit/passes/lower_tuples.cpp:109)
frame #0: std::function<std::string ()>::operator()() const + 0x11 (0x7f27f31b3fe1 in /home/train/.local/lib/python3.5/site-packages/torch/lib/libc10.so)
frame #1: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x2a (0x7f27f31b3dfa in /home/train/.local/lib/python3.5/site-packages/torch/lib/libc10.so)
frame #2: <unknown function> + 0x6da2e1 (0x7f27def7f2e1 in /home/train/.local/lib/python3.5/site-packages/torch/lib/libtorch.so.1)
frame #3: <unknown function> + 0x6da534 (0x7f27def7f534 in /home/train/.local/lib/python3.5/site-packages/torch/lib/libtorch.so.1)

搜索后发现和这个错误类似.
DataParallel移除即可.

model = mobilenetv2()
model = torch.nn.DataParallel(model).cuda()

改为

model = mobilenetv2().cuda()

问题记录贴

原文:https://www.cnblogs.com/sdu20112013/p/13299118.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!