mindspore.parallel.merge_pipeline_strategys

mindspore.parallel.merge_pipeline_strategys(src_strategy_dirs, dst_strategy_file)

汇聚所有流水线并行子图的切分策略文件到目的文件。

说明

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

参数：

• src_strategy_dirs (str) - 包含所有流水线并行的子图的切分策略文件的目录，切分策略文件由 mindspore.parallel.auto_parallel.AutoParallel.save_param_strategy_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.parallel.merge_pipeline_strategys("./src_strategy_dir", "./dst_strategy.ckpt")