search

longcw / pytorch2caffe

Convert PyTorch model to Caffemodel
541 stars 151 forks source link

'MaxPool2DBackward' object has no attribute 'stride' #8

Open niaoyu opened 6 years ago

niaoyu commented 6 years ago

Some trouble happen when i use the tool for squeezenet.

First , it looks like names of some layers_grad had been changed,: MaxPool2dBackward=>MaxPool2DBackward

Then, after added the new "MaxPool2DBackward", it caused another error,seems like: "'MaxPool2DBackward' object has no attribute 'stride" or "'MaxPool2DBackward' object has no attribute 'kernel_size"

corresspoding to the code: pooling_param['kernel_size'] = func.kernel_size[0] pooling_param['stride'] = func.stride[0]

Does anyone could help?

wangbofei11 commented 6 years ago

hello,I also meet this problem, when use pytorch with version>=0.3.0, this error happend, the grad_fn's name changed and it's param also changed. until now, I have not found solutions,if you can find a solution,please told me . tks!

Jongchan commented 6 years ago

Same here. Also facing the same problem here.