xuyan1115 / pytorch2caffe

Convert PyTorch model to Caffemodel
8 stars 1 forks source link

AttributeError: 'MaxPool2DBackward' object has no attribute 'kernel_size' #2

Open lzhaozi opened 6 years ago

lzhaozi commented 6 years ago

My python env is Python 2.7.14 and PyTorch 0.3.1. I try to convert my model, but face this problem: AttributeError: 'MaxPool2DBackward' object has no attribute 'kernel_size'

Is it because this repo does not support PyTorch 0.3.1?

xuyan1115 commented 6 years ago

yep, the interfaces in PyTorch 0.3.0 are different from other versions.