SNPE MaxPerGPUSize 注意事项

SNPE根据不同的硬件设备支持的GPU每层卷积运算的能力不同,具体如下:
A330平台:8192
A430,A530平台: 16384
H或者W乘以通道数不能大于上面的上限,如51251232 size的feature map 刚好是A430,A530设备的最大量级。

值得注意: 我们设计网络时,容易忽略padding带来的计算数据变大问题。比如,输入51251232数据,按说刚刚好可以计算,但是如果卷积有padding操作,则有可能变成51351332,数值超过计算上限了,导致结果和PC端的不一样,这一点很容易忽略。

建议: 尽量不要极限最大程度的使用SNPE,很容易忽略某个细节,导致数据越界计算不正确。此外,SNPE有时候会提醒你有可能越界,有时候不会提醒你,所以,尽量不要把网络设计的非常极限使用GPU。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值