efficientnet转onnx遇到的问题

efficientnet转换出现问题

问题:RuntimeError: Exporting the operator silu to ONNX opset version 11 is not supported

解决方案:

1.更新pytorch=1.8.1支持相应的操作;
2.使用自定义模块处理,过程比较复杂,可参考官网;

问题:Driver error

解决方案:升级显卡驱动

你可能感兴趣的:(自定义op,pytorch)