mindspore.merge_pipeline_strategys

查看源文件
mindspore.merge_pipeline_strategys(src_strategy_dirs, dst_strategy_file)[源代码]

流水线并行模式下,汇聚所有流水线并行子图的切分策略文件。关于更多分布式Checkpoint转换的细节,请参考:模型转换

说明

src_strategy_dirs必须包含所有流水线并行的子图的切分策略文件。

参数:
  • src_strategy_dirs (str) - 包含所有流水线并行的子图的切分策略文件的目录,切分策略文件由mindspore.set_auto_parallel_context(strategy_ckpt_save_file)接口存储得到。

  • dst_strategy_file (str) - 保存汇聚后的切分策略的文件路径。

异常:
  • NotADirectoryError - src_strategy_dirs 不是一个目录。

样例:

>>> import mindspore as ms
>>> # src_strategy_dir/stra0.ckpt, src_strategy_dir/stra1.ckpt ... src_strategy_dir/stra127.ckpt
>>> ms.merge_pipeline_strategys("./src_strategy_dir", "./dst_strategy.ckpt")