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

关于GAP的问题 #88

Open CangHaiQingYue opened 4 years ago

CangHaiQingYue commented 4 years ago

大佬好！ 我的网络中有nn.AdaptiveAvgPool2d(1)这个一层，正常情况下，应该调用_adaptive_avg_pool2d()函数，但是我这边定位到，它调用了_pool()函数，导致后面的一些列报错，这个问题该怎么fix，请指点下~

riskeverything commented 3 years ago

大佬好！ 我的网络中有nn.AdaptiveAvgPool2d(1)这个一层，正常情况下，应该调用_adaptive_avg_pool2d()函数，但是我这边定位到，它调用了_pool()函数，导致后面的一些列报错，这个问题该怎么fix，请指点下~

请问，你解决了这个问题了嘛？我用SE同样遇到了这个问题。

CangHaiQingYue commented 3 years ago

大佬好！ 我的网络中有nn.AdaptiveAvgPool2d(1)这个一层，正常情况下，应该调用_adaptive_avg_pool2d()函数，但是我这边定位到，它调用了_pool()函数，导致后面的一些列报错，这个问题该怎么fix，请指点下~

请问，你解决了这个问题了嘛？我用SE同样遇到了这个问题。

你好，我好像没有解决~

riskeverything commented 3 years ago

大佬好！ 我的网络中有nn.AdaptiveAvgPool2d(1)这个一层，正常情况下，应该调用_adaptive_avg_pool2d()函数，但是我这边定位到，它调用了_pool()函数，导致后面的一些列报错，这个问题该怎么fix，请指点下~

请问，你解决了这个问题了嘛？我用SE同样遇到了这个问题。

你好，我好像没有解决~

我的问题解决了，最后发现报错的根本原因出在了我用到了SE模块，这个最后有一个转换是11和nn的tensor相乘，导致过不去。adaavgpool是可以转的。

CangHaiQingYue commented 3 years ago

@riskeverything 谢谢