NPU_LOOP_SIZE

Description

Sets the number of iterations per loop offloaded to the NPU in the TensorFlow 2.6.5 training and online inference scenarios.

Example

export NPU_LOOP_SIZE=32

Restrictions

• This variable should be set before the import npu_device operation.
• This environment variable applies only to the scenario where the TensorFlow 2.6.5 network training or online inference is performed on the Ascend platform.

Applicability

Atlas Training Series Product