原因:海思平台不支持maxpool中ceil_mode=false,
resnet中的maxpool默认为false。
torch.nn — PyTorch 1.11.0 documentation
解决办法:
resnet maxpool只有一处:
1、改变输入
改变输入大小,使得不受ceil_mode影响。
pool计算公式:
如:
原因:海思平台不支持maxpool中ceil_mode=false,
resnet中的maxpool默认为false。
torch.nn — PyTorch 1.11.0 documentation
解决办法:
resnet maxpool只有一处:
1、改变输入
改变输入大小,使得不受ceil_mode影响。
pool计算公式:
如: