TensorFlow 1.15训练场景下,用于设置NPU上总的训练步数。取值为int类型,无默认值。
export TOTAL_STEP=10000
否
该环境变量仅适用于TensorFlow 1.15网络在昇腾平台执行训练的场景。
Atlas 训练系列产品
Atlas A2 训练系列产品