search

xxradon / PytorchToCaffe

Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs
MIT License
783 stars 224 forks source link

无法转换 Avgpool2d #46

Closed kyleleey closed 4 years ago

kyleleey commented 4 years ago

不论在网络结构中使用mean还是avgpool2d函数，转换后都会转为caffe中的'Reduction'层，但我后期的部署中不支持这种层。不过PytorchToCaffe的源码中好像有转为caffe中poo层的代码，请问我应该如何在pytorch中应用avgpool功能？我的pytorch版本是1.2.0

xxradon commented 4 years ago

now master branch supoort avgpool2d

kyleleey commented 4 years ago

I see, thank you for that