当前位置：网站首页>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

边栏推荐

猜你喜欢

随机推荐