mindspore.amp Cell管理

qq_33007875 2022-11-15 11:02:00

mindspore.amp

Cell管理

mindspore.amp.DynamicLossScaleManager

动态调整损失缩放系数的管理器,继承自 mindspore.amp.LossScaleManager 。

mindspore.amp.LossScaleManager

使用混合精度时,用于管理损失缩放系数(loss scale)的抽象类。

mindspore.amp.FixedLossScaleManager

损失缩放系数不变的管理器,继承自 mindspore.amp.LossScaleManager 。

mindspore.amp.build_train_network

构建混合精度训练网络。

函数式

mindspore.amp.DynamicLossScaler

动态调整损失缩放系数的管理器。

mindspore.amp.StaticLossScaler

Static Loss scale类。

mindspore.amp.LossScaler

使用混合精度时,用于管理损失缩放系数(loss scaler)的抽象类。

mindspore.amp.auto_mixed_precision

对Cell进行自动混合精度处理。

mindspore.amp.all_finite

检查inputs是否是有效值(无溢出)。

...全文
58 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复

164

社区成员

发帖
与我相关
我的任务
社区描述
AI技术分享以及交流
人工智能 其他
社区管理员
  • 跳楼梯企鹅
  • 平凡的人1
  • 微枫Micromaple
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧