ttp_init_controller
接口功能
RC2版本使用，初始化MindIO TFT Controller模块。
接口格式
mindio_ttp.ttp_init_controller(rank: int, world_size: int, replica: int, enable_local_copy: bool)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
rank |
必选 |
当前执行训练任务的NPU ID。 |
大于或等于0。 |
world_size |
必选 |
整个集群参与训练任务的卡数。 |
大于0。 |
replica |
可选 |
适配老版本的参数，实际无效。 |
int。 |
enable_local_copy |
可选 |
是否启用local copy。优化器更新前，先对优化器做一次备份。 |
|
返回值
无
父主题： API接口参考