search

xiezheng-cs / PytorchToCaffe

A tool for PytorchToCaffe，fork form https://github.com/xxradon/PytorchToCaffe
MIT License
11 stars 2 forks source link

FeatherNet #3

Open ly0303521 opened 5 years ago

ly0303521 commented 5 years ago

你好，我现在能将FeatherNet的pytorh模型生成caffemodel和prototxt，但是prototxt好像不对，前向的时候会报错。view2维度是16x1x1，但是add1的维度是16x56x56，这两个mul1 做运算的时候会报错 2019-09-18 10-19-29屏幕截图

ly0303521 commented 5 years ago

刚刚参照你的文档说明“A method to convert SELayer from pytorch to caffe” ，修改网络后运行验证程序 “FaceFeatherNetB_verify_deploy.py”， 没有报错，但是pytorch模型和caffe模型输出差好多 output_diff: 2146.886719， 应该哪里还是不对

ly0303521 commented 5 years ago

问题找到了，原始的FeatherNet 激活函数用了 relu6， 转换工具并不支持， 现在换成了relu，没有问题

xiezheng-cs commented 5 years ago

@ly0303521 是的，caffe版本不支持 relu6, 但可以用pytorch2onnx，可以转