![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
想学CNN的Shuai
这个作者很懒,什么都没留下…
展开
-
IndexError: _Map_base::at
由pytorch到onnx转换efficientnet,onnx引用转换好的模型时报错:line 55, in optimizeoptimized_model_str = C.optimize(model_str, passes)IndexError: _Map_base::at解决方法:在export时添加 keep_initializers_as_inputs 参数:torch.o...原创 2019-11-15 13:51:07 · 4318 阅读 · 2 评论 -
RuntimeError: ONNX export failed: Couldn't export Python operator SwishImp
efficientnet pytorch转onnx 时,执行其官方脚本:https://github.com/lukemelas/EfficientNet-PyTorch#loading-pretrained-modelsmodel = EfficientNet.from_name(model_name=‘efficientnet-b0’)torch.onnx.export(model, t...原创 2019-11-15 13:45:58 · 3844 阅读 · 0 评论