nnie模型(海思平台)平台转模型改变输入消除ceil_mode影响

原因:海思平台不支持maxpool中ceil_mode=false, resnet中的maxpool默认为false。 https://pytorch.org/docs/stable/nn.html   解决办法: resnet  maxpool只有一处: 改变输入大小,使得不受ceil_mode影响。 pool计算公式: 如:输入为384, 卷积后为192 (192+2*1+1*(3-1)-1)
相关文章
相关标签/搜索