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)
相關文章
相關標籤/搜索