mindspore.parallel.merge_pipeline_strategys

mindspore.parallel.merge_pipeline_strategys(src_strategy_dirs, dst_strategy_file)[source]

Aggregate the sharding strategy files of all pipeline parallel subgraphs to the destination file.

Note

Strategy file of each pipeline stage should be included in src_strategy_dirs.

Parameters

src_strategy_dirs (str) – The directory of strategy files including all pipeline stage which is saved by mindspore.parallel.auto_parallel.AutoParallel.save_param_strategy_file().
dst_strategy_file (str) – The file merged strategy to save.

Raises

NotADirectoryError – src_strategy_dirs is not a directory.

Examples

>>> 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")