问题描述:
【功能模块】
优化器求梯度时报错
【操作步骤&问题现象】
1、按照流程构建网络,各部分单元测试跑通
2、在pynative下运行模型,报错,应该是优化器求梯度时出现问题,但不知道具体是哪里的问题,如何修改
【截图信息】
解答:
这个报错表示StridedSliceGrad算子编译失败了,原因是算子参数strides[1]的值有误,应该是1,但是当前是-1,你网络里有调用StridedSlice吗?检查下参数是否正确,应该是因为使用了[::-1]这样的操作。
问题描述:
【功能模块】
优化器求梯度时报错
【操作步骤&问题现象】
1、按照流程构建网络,各部分单元测试跑通
2、在pynative下运行模型,报错,应该是优化器求梯度时出现问题,但不知道具体是哪里的问题,如何修改
【截图信息】
解答:
这个报错表示StridedSliceGrad算子编译失败了,原因是算子参数strides[1]的值有误,应该是1,但是当前是-1,你网络里有调用StridedSlice吗?检查下参数是否正确,应该是因为使用了[::-1]这样的操作。