下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

NPUStrategy构造函数

功能说明

NPUStrategy类的构造函数。NPUStrategy继承了tf.distribute.Strategy类,可以调用基类的原生接口,用于在NPU环境中实现分布式训练。

函数原型

def __init__(self)

参数说明

返回值

NPUStrategy类对象。

调用示例

构建NPUStrategy实例,在NPU环境实现分布式训练:

1
2
3
4
5
from npu_bridge.npu_init import *
...
strategy = npu_strategy.NPUStrategy()
#使用strategy实现分布式训练。使用方式与tf.distribute.Strategy相同
...
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词