pytorch2caffe doesn't work for the last two layyers's convertion about pytorch2caffe HOT 5 CLOSED

@dzyjjpy Hello, i am also trying to convert a Head Pose Estimation (Hopenet) model trained using pytorch to caffe by this repository. Can you share some advice how to use this repository to do so ?

from pytorch2caffe.

@dzyjjpy @trohit920 Thanks for interested in this repo and I am very glad this is helpful for you. What's the type of your last two layers? Or you can provide me a simple model for debugging. I will try to find the reason when I have free time.

from pytorch2caffe.

@longcw I am working on converting Hopenet model Link to caffe. You can check model file and download the model from this link model. Plese take a look. Thanks in advance.

from pytorch2caffe.

@longcw Hi , I have convert successfully before. thanks for your concern

from pytorch2caffe.

@trohit920 after network initialization, call the pytorch2caffe function to convert.
pay attention to the output layers, as it has three branch, so you need to edit some code in pytorch2caffe.py

from pytorch2caffe.