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)