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