问题描述:
【功能模块】
mindspore1.2 cpu
【操作步骤&问题现象】
1、在计算loss值时可以正常计算得到结果
2、在把loss网络添加到trainonestepcell之后,就不能正常计算,报错如下:
我的模型里并不包含这一算子。但是他会报找不到这个算子的错误。
【截图信息】
具体的withlosscell实现:
具体的trainOnsestepcell实现:
希望能得到讲解,谢谢!
【日志信息】(可选,上传日志内容或者附件)
解决方案:
目前RowTensor 不支持pynative