mindspore.get_ckpt_path_with_strategy
- mindspore.get_ckpt_path_with_strategy(cur_ckpt_path, cur_strategy_path)[源代码]
从当前卡的所有存在备份关系的checkpoint文件中找到可用的checkpoint文件。 该接口假定文件名中存在rank_{rank_id}的字符串用于区分不同的checkpoint文件。如果不存在该字符串并且传入路径可用,则返回传入路径,否则返回None。
说明
这个API必须在集群初始完成后调用,因为接口内部需要获取集群信息。
- 参数:
cur_ckpt_path (str) - 当前卡需要的checkpoint的文件路径。
cur_strategy_path (str) - 当前卡的strategy文件。
- 返回:
str, 如果找到可用checkpoint文件,返回该路径。
None,如果未找到可用checkpoin文件,返回None。
样例:
>>> import mindspore as ms >>> from mindspore.communication import init >>> from mindspore import get_ckpt_path_with_strategy >>> ms.set_context(mode=ms.GRAPH_MODE) >>> ms.set_auto_parallel_context(parallel_mode=ms.ParallelMode.DATA_PARALLEL, gradients_mean=True) >>> init() >>> ckpt_file= "./rank_5/iteration-1_40.ckpt" >>> strategy_file = "./src_pipeline_strategys/src_strategy_5.ckpt" >>> ckpt_file_new = get_ckpt_path_with_strategy(ckpt_file, strategy_file) >>> print(ckpt_file_new)