打包onnx遇到了一个bug,意思是我的某个节点的运算时维度不对,通过netron查看了节点,翻看了代码以后是没有问题的。
import netron model="model.onnx" netron.strat(model)
最后的解决方式是,卸载pyroch,重装torch和torchvision,pillow和其他依赖包,我的模型里还需要更新pip install --upgrade charset-normalizer。
今天又是神奇的一天