TensorFlow 1.15训练场景下,若通过“experimental_accelerate_train_mode”参数或者“accelerate_train_mode”参数触发了训练加速功能,可通过此环境变量设置NPU上总训练步数。
- 若训练脚本中使用session配置,关于该环境变量使用场景的详细介绍可参见session配置中的参数的介绍。
- 若训练脚本中使用NPURunConfig配置,关于该环境变量使用场景的详细介绍可参见npu.npu_config中ExperimentalConfig构造函数的参数的介绍。
该环境变量取值为int类型,无默认值。
[object Object]
该环境变量仅适用于TensorFlow 1.15网络在昇腾平台执行训练的场景。
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 训练系列产品