RuntimeError: ONNX export failed: Couldn‘t export operator aten::adaptive_avg_pool2d

在对ResNet-101模型pth文件转换成onnx文件时,出现问题:
RuntimeError: ONNX export failed: Couldn‘t export operator aten::adaptive_avg_pool2d_第1张图片

我的解决方法是,将模型中的最后的池化层注释掉了:

在这里插入图片描述
改为:
在这里插入图片描述
这样就不报错了

你可能感兴趣的:(python,深度学习)