search

longcw / pytorch2caffe

Convert PyTorch model to Caffemodel
541 stars 151 forks source link

convert KeyError: 'MkldnnConvolutionBackward' #35

Closed yueyihua closed 4 years ago

yueyihua commented 4 years ago

[root@greenet tools]# python torch2caffe.py Traceback (most recent call last): File "torch2caffe.py", line 29, in pytorch2caffe(input_var, output_var, './tc-pytorch2caffe.prototxt', './tc-pytorch2caffe.caffemodel') File "/home/yyh/tools/pytorch2caffe/pytorch2caffe.py", line 37, in pytorch2caffe net_info = pytorch2prototxt(input_var, output_var) File "/home/yyh/tools/pytorch2caffe/pytorch2caffe.py", line 325, in pytorch2prototxt add_layer(output_var.grad_fn) File "/home/yyh/tools/pytorch2caffe/pytorch2caffe.py", line 153, in add_layer top_name = add_layer(u[0]) File "/home/yyh/tools/pytorch2caffe/pytorch2caffe.py", line 153, in add_layer top_name = add_layer(u[0]) File "/home/yyh/tools/pytorch2caffe/pytorch2caffe.py", line 153, in add_layer top_name = add_layer(u[0]) [Previous line repeated 58 more times] File "/home/yyh/tools/pytorch2caffe/pytorch2caffe.py", line 165, in add_layer layer['type'] = layer_dict[parent_type] KeyError: 'MkldnnConvolutionBackward'

yueyihua commented 4 years ago

The same as #25