CreateGPUKernel] gpu kernel op[MaximumGrad-op1014] Resize failed

网络反向传播,求导过程中报错

【操作步骤&问题现象】

【截图信息】

具体报错语句:

【日志信息】(可选,上传日志内容或者附件)

报错信息

在截图信息展示的代码中,报错语句调用的参数 *args包含的tensor最高维度是4维,并没有报错信息说的5维输入,这个错误该如何处理呢?

可以设置context.set_context(save_graphs=True)保存IR,从而可以看到算子的shape信息。如果你的脚本里没有显式调用MaximumGrad算子的话,那应该是Maximum或者ClipByNorm算子的反向计算逻辑中调用到的。

如果存在扩维的操作,那么Maximum接受到的输入就有可能是高维的了,从而在反向计算过程中也是高维输入。目前了解到MindSpore master分支代码已经取消了MaximumGrad输入<= 4D的限制,当前限制是<=7D即可。对应的源码(感兴趣的话可以查看该源码文件的最新修改):https://gitee.com/mindspore/mindspore/blob/master/mindspore/ccsrc/plugin/device/gpu/kernel/math/broadcast_grad_gpu_kernel.cc 你需要自行编译MindSpore master代码或者等1.8正式版本发布。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值