ExperimentalConfig构造函数
功能说明
ExperimentalConfig类的构造函数,调试功能扩展参数,后续版本可能会存在变动,不支持应用于商用产品中。
函数原型
def __init__(self, accelerate_train_mode=="fast|step|0.9", …… )
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
accelerate_train_mode |
输入 |
针对超过1小时以上的训练场景,开发者可以通过此配置触发训练加速,提升训练性能。 软件内部会根据开发者配置的加速类型、加速触发模式以及低精度训练流程占比,对相应比例的训练流程降精度编译运行,剩余的训练流程仍按照原始精度编译运行。
该配置项取值类型为字符串,由“|”符号分割为三个字段,例如:fast|step|0.9。
配置示例:
需要注意:
|
返回值
返回ExperimentalConfig类对象,作为NPURunConfig的参数传入。
约束说明
无。
调用示例
1 2 3 4 5 6 |
from npu_bridge.npu_init import * from npu_bridge.estimator.npu.npu_config import ExperimentalConfig ... experimental_config=ExperimentalConfig(accelerate_train_mode="fast|step|0.9") session_config=tf.ConfigProto(allow_soft_placement=True) config = NPURunConfig(experimental_config=experimental_config, session_config=session_config) |