文档
注册
评分
提单
论坛
小AI

STEP_NOW

功能描述

TensorFlow 1.15训练场景下,若通过“experimental_accelerate_train_mode”参数或者“accelerate_train_mode”参数触发了训练加速功能,可通过此环境变量设置NPU上当前的执行步数。

  • 若训练脚本中使用session配置,关于该环境变量使用场景的详细介绍可参见TF Adapter API(1.x)中“session配置 > session配置参数说明 > 试验参数”中的“experimental_accelerate_train_mode”参数的介绍。
  • 若训练脚本中使用NPURunConfig配置,关于该环境变量使用场景的详细介绍可参见TF Adapter API(1.x)中“npu_bridge.estimator.npu.npu_config > ExperimentalConfig构造函数 ”中的“accelerate_train_mode”参数的介绍。

该环境变量取值为int类型,无默认值。

配置示例

export STEP_NOW=100

使用约束

该环境变量仅适用于TensorFlow 1.15网络在昇腾平台执行训练的场景。

支持的型号

Atlas 训练系列产品

Atlas A2 训练系列产品

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词