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相同 ... |