Error to convert a model from IR to Pytorch

microsoft / MMdnn

MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.

MIT License

5.8k stars 965 forks source link

Error to convert a model from IR to Pytorch #941

Open luismadhe opened 1 year ago

luismadhe commented 1 year ago

I got this error when I try to pass a model from IR to Pytorch:

RuntimeError: output with shape [96] doesn't match the broadcast shape [1, 1, 1, 96]

The IR files comes frome a caffe model, then the complete procces is to convert a caffe model to pytorch.