NPU_LOOP_SIZE
功能描述
TensorFlow 2.6.5训练与在线推理场景下,用于设置NPU上循环下沉的次数。
配置示例
export NPU_LOOP_SIZE=32
使用约束
- 该变量需要在import npu_device前设置。
- 该环境变量仅适用于TensorFlow 2.6.5网络在昇腾平台执行训练或在线推理的场景。
支持的型号
Atlas 训练系列产品
Atlas 推理系列产品
Atlas A2 训练系列产品
父主题: TF Adapter
TensorFlow 2.6.5训练与在线推理场景下,用于设置NPU上循环下沉的次数。
export NPU_LOOP_SIZE=32
Atlas 训练系列产品
Atlas 推理系列产品
Atlas A2 训练系列产品
当前产品无相关内容
未找到相关内容,请尝试其他搜索词