参数名 |
描述 |
---|---|
graph_run_mode |
图执行模式,取值:
配置示例: config = NPURunConfig(graph_run_mode=1) |
session_device_id |
当用户需要将不同的模型通过同一个训练脚本在不同的Device上执行,可以通过该参数指定Device的逻辑ID。 通常可以为不同的图创建不同的Session,并且传入不同的session_device_id,该参数优先级高于ASCEND_DEVICE_ID。 配置示例: config0 = NPURunConfig(..., session_device_id=0, ...) estimator0 = NPUEstimator(..., config=config0, ...) ... config1 = NPURunConfig(..., session_device_id=1, ...) estimator1 = NPUEstimator(..., config=config1, ...) ... config7 = NPURunConfig(..., session_device_id=7, ...) estimator7 = NPUEstimator(..., config=config7, ...) ... |
distribute |
通过PS-Worker架构进行分布式训练时,用于传入ParameterServerStrategy对象。 配置示例: config = NPURunConfig(distribute=strategy) |