KeyError: 'ExpandBackward' about pytorch2caffe HOT 12 OPEN

Let pytorch=0.2, torchvision=0.1.8, the problem will be solved!
I don't kown why, maybe newer pytorch use 'ExpandBackward' to name layer, but older pytorch do not use. And the layer_dict in pytorch2caffe.py do not have 'ExpandBackward'

from pytorch2caffe.

@wyfdodogod I think you needn't retrain in torch 2, you should understand the meaning of ExpandBackward and add this key to layer_dict , then you should add some operater in function pytorch2prototxt, just like elif parent_type == 'BatchNormBackward':

from pytorch2caffe.

I have met same problems. I guess the conversion code has not support 'ExpandBackward' currently.

from pytorch2caffe.

met same problems too!

from pytorch2caffe.

same problem， can anybody give some tips？

from pytorch2caffe.

met same problems too!

from pytorch2caffe.

does that mean any model trained in torch version 3 and above need to be retrained in torch 2 and then safe to be converted to caffe

from pytorch2caffe.

@autocyz ty for the explaination, as you mentioned using torch 0.2 and torchvision 0.1.8, i solved the problem more specifically torch0.2 post1 using post2 and 3 might raise error causing python crash at the time caffe is imported.

from pytorch2caffe.

@wangxc2006 did you retrain the model under torch 0.2?

from pytorch2caffe.

Any plan to support 0.4.x ?

from pytorch2caffe.

@autocyz @longcw met same issue when running demo.py.
when i try to install 0,2.0 by pip install torch==0.2.0 or pip isntall torch-0.2.0.post********.whl
(I try install it based CUDA 9.2 or CUDA 8.0)
errors displays:
is not a supported wheel on this platform
so strange. dose anyone run demo,py sucessfully on torch==0.3.1 and torchvision==0.2.0
I trained model on the env above. thanks

from pytorch2caffe.

@autocyz Like you said, "should understand the meaning of ExpandBackward and add this key to layer_dict , then you should add some operater in function pytorch2prototxt, just like elif parent_type == 'BatchNormBackward':"
But i don't know ‘ExpandBackward’ means. So how should it be modified?

from pytorch2caffe.