mmtoir -f pytorch -w imagenet_vgg16.pth --inputShape 3,224,224 -p ./toir_vgg16

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

mmtoir -f pytorch -w imagenet_vgg16.pth --inputShape 3,224,224 -p ./toir_vgg16 #657

Open zhao181 opened 5 years ago

zhao181 commented 5 years ago

I am using the following command to convert pytorch model to ir, mmtoir -f pytorch -w imagenet_vgg16.pth --inputShape 3,224,224 -p ./toir_vgg16

No errors are complained, but also no output files are created. Can anyone help offer a solution?

JiahaoYao commented 5 years ago

Hi, could you give more information? Or please check this page for pytorch conversion.