NPUStrategy Constructor

Applicability

Description

Constructs an object of class NPUStrategy NPUStrategy inherits the tf.distribute.Strategy class and can call the native APIs of the base class to implement distributed training in the NPU environment.

Prototype

1
2

class NPUStrategy(distribute_lib.StrategyV1):
    def __init__(self, device="/cpu:0")

Parameters

Returns

An object of the NPUStrategy class

Example

Build an instance of NPUStrategy to implement distributed training with NPU environments.

1
2
3
4
5

from npu_bridge.npu_init import *
...
strategy = npu_strategy.NPUStrategy()
# Use strategy to implement distributed training. The usage is the same as that of tf.distribute.Strategy.
...