当前位置:网站首页>1.3版本自定义TrainOneStepCell报错

1.3版本自定义TrainOneStepCell报错

2022-08-11 08:52:00 小乐快乐

环境:ASCEND 910 mindspore 1.3

【功能模块】

自定义TrainOneStepCell进行梯度裁剪,使用的是model zoo里面bert的代码,1.2版本正常,1.3版本报如下面图片所示错误

在官网中有自定义TrainOneStepCell教程,建议可以先参考检查下

https://www.mindspore.cn/tutorials/zh-CN/r1.5/intermediate/custom/train.html

 

原网站

版权声明
本文为[小乐快乐]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_45666880/article/details/126270315