I tried sample code in the README and I've got below error message.
starting to transfrom net vgg16
Traceback (most recent call last):
File "example.py", line 16, in
name='vgg16')
File "/home/jihoonk/workspace/pytorch2caffe/nn_tools/pytorch_to_caffe.py", line 205, in pytorch_to_caffe
trans_to_protobuf(net, input_var, output_var, name)
File "/home/jihoonk/workspace/pytorch2caffe/nn_tools/pytorch_to_caffe.py", line 193, in trans_to_protobuf
add_layer(output_var.grad_fn)
File "/home/jihoonk/workspace/pytorch2caffe/nn_tools/pytorch_to_caffe.py", line 51, in add_layer
top_name = add_layer(u[0])
File "/home/jihoonk/workspace/pytorch2caffe/nn_tools/pytorch_to_caffe.py", line 62, in add_layer
parent_name = layer_dict[parent_type] + str(layer_id)
KeyError: 'ExpandBackward'
The old version of the pytorch2caffe is not supporting the pytorch 0.3.0. I write a new version of the convertor for new version of the 0.3.0. Wish this new version can convert your model successfully.
I tried sample code in the README and I've got below error message.
My pytorch version is the latest one, http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl