mindspore.restore_group_info_list

查看源文件
mindspore.restore_group_info_list(group_info_file_name)[源代码]

从 group_info_file_name 指向的文件中提取得到通信域的信息，在该通信域内的所有设备的checkpoint文件均与存储 group_info_file_name 的设备相同，可以直接进行替换。通过配置环境变量GROUP_INFO_FILE以在编译阶段存储下该通信域信息，例如”export GROUP_INFO_FILE=/data/group_info.pb”。

参数：

  • group_info_file_name (str) - 保存通信域的文件的名字。

返回：

List，通信域列表。

异常：

  • ValueError - 通信域文件格式不正确。

  • TypeError - group_info_file_name 不是字符串。

样例：

>>> import mindspore as ms
>>> ms.restore_list = restore_group_info_list("./group_info.pb")