mindspore.parallel.build_searched_strategy
- mindspore.parallel.build_searched_strategy(strategy_filename)[源代码]
网络中每个参数的构建策略,用于分布式推理的场景。
- 参数:
strategy_filename (str) - 策略文件的名称。
- 返回:
Dict,key为参数名,value为该参数的切片策略。
- 异常:
ValueError - 策略文件不正确。
TypeError - strategy_filename 不是字符串类型。
- 支持平台:
Ascend
样例:
>>> from mindspore.parallel import build_searched_strategy >>> strategy = build_searched_strategy("./strategy_train.ckpt")