how to convert my selfdefine pytorch model to caffe model.

hahnyuan / nn_tools

Neural Network Tools: Converter and Analyzer. For caffe, pytorch, draknet and so on.

MIT License

355 stars 63 forks source link

how to convert my selfdefine pytorch model to caffe model. #16

Open yongjingli opened 5 years ago

yongjingli commented 5 years ago

hi,I want to know how to convert selfdefine pytorch model to caffe model, thanks a lot.

yongjingli commented 5 years ago

hi, hahnyuan @hahnyuan I want to convert the LPRnet to caffe model, the LPRnet comes from here: https://github.com/sirius-ai/LPRNet_Pytorch/blob/master/model/LPRNet.py. But I suffer the below problem, I want to know what's the problem:

yongjingli commented 5 years ago

hi, hahnyuan @hahnyuan , and now I suffer another problem. And the problem is that I use different stride=(1, 2) in max pooling layer, which is not support in pytorch2caffe: