TOTAL_STEP

Description

In TensorFlow 1.15 training scenarios, if training acceleration is enabled via the experimental_accelerate_train_mode or accelerate_train_mode parameter, this environment variable can be used to set the total number of training steps executed on the NPU.

• If the session configuration is used in the training script, see the description of the experimental_accelerate_train_mode parameter in "Session Configuration > Session Configuration Parameters > Experiment Parameters" in TF Adapter API (1.x).
• If NPURunConfig is used in the training script, see the description of the accelerate_train_mode parameter in "npu_bridge.estimator.npu.npu_config > ExperimentalConfig Constructor" in TF Adapter API (1.x).

The value of this environment variable is of the int type. There is no default value.

Example

export TOTAL_STEP=10000

Constraints

This environment variable applies only to the scenario where the TensorFlow 1.15 network is trained on the Ascend platform.

Applicability

Atlas 350 Accelerator Card

Atlas A3 training product/Atlas A3 inference product

Atlas A2 training product/Atlas A2 inference product

Atlas training product