mindspore.rank_list_for_transform

mindspore.rank_list_for_transform(rank_id, src_strategy_file=None, dst_strategy_file=None)[源代码]

在对分布式Checkpoint转换的过程中，获取为了得到目标rank的Checkpoint文件所需的源Checkpoint文件rank列表。关于更多分布式Checkpoint转换的细节，请参考：模型转换。

参数：

rank_id (int) - 待转换得到的Checkpoint的rank号。
src_strategy_file (str) - 源切分策略proto文件名，由 mindspore.set_auto_parallel_context(strategy_ckpt_save_file) 接口存储下来的文件。当其为 None 时，表示切分策略为不切分。默认值： None 。
dst_strategy_file (str) - 目标切分策略proto文件名，由 mindspore.set_auto_parallel_context(strategy_ckpt_save_file) 接口存储下来的文件。当其为 None 时，表示切分策略为不切分。默认值： None 。

返回：

转换得到rank_id的分布式Checkpoint所需要的卡号列表。

异常：

ValueError - src_strategy_file 或者 dst_strategy_file 不是正确的切分策略proto文件。
TypeError - src_strategy_file 或者 dst_strategy_file 不是字符串。
TypeError - rank_id 不是一个整数。

样例：

>>> import mindspore as ms
>>> rank_id = 0
>>> rank_list = ms.rank_list_for_transform(rank_id, "./src_strategy.ckpt", "./dst_strategy.ckpt")
>>> checkpoint_files_map = {}
>>> for rank in rank_list:
...     checkpoint_files_map[rank] = "./pangu{}-100_2.ckpt".format(rank)