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.78k stars 968 forks source link

load higher version pytorch model error #924

Open RiceZ opened 3 years ago

RiceZ commented 3 years ago

Hi, there are some errors when I transfer my pytorch model to caffe model. I run it in docker.

image

I guess the reason is that the versions of torch are not matched. docker's torch version is 0.4.0 and mine is 1.7.1. I try to load this pytorch model and get the same errors:

image

What should I do now? Because there will be lots of errors if I train model using torch 0.4.0 with my current code. Btw, What is the plan of docker update?

thank you very much!