/pytorch/aten/src/ATen/native/cuda/ScatterGatherKernel.cu:115: operator(): block:

本文分享了如何在GPU上有效处理神经网络模型错误,如数组超限问题,通过锚点比例调整和focal loss中gamma参数一致性来举例。同时提及了Cuda错误信息的处理策略,以及参考了博主关于IndexKernel和ScatterGatherKernel的调试经验。

低功耗蓝牙项目,需要一块懂省电的板

思澈 SF32LB52 芯片,BLE 协议栈深度优化,上手即开发

GPU上运行神经网络模型,是使用多线程并行加速的,如果出错,真正的错误信息很容易被cuda的错误信息淹没,所以难以看到错误信息,一般引起问题的原因是数组等超限,定位到具体是anchor_ratios=[0.2, 0.5, 1.0, 2.0, 5.0]修改成默认anchor_ratios=[0.5, 1.0, 2.0]就可以啦。

比如,focalloss中gamma数组个数和数据类别数不一致也会报错:

/pytorch/aten/src/ATen/native/cuda/ScatterGatherKernel.cu:115: operator(): block:

将维度修改成一致即可


参考:

pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:53: lambda [](int)->auto::operator()(int)->auto: bl_Wanderer001的博客-CSDN博客

低功耗蓝牙项目,需要一块懂省电的板

思澈 SF32LB52 芯片,BLE 协议栈深度优化,上手即开发

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值