mindspore.build_searched_strategy

mindspore.build_searched_strategy(strategy_filename)[源代码]

构建网络中每个参数的策略,用于分布式推理。

参数:
  • strategy_filename (str) - 策略文件的名称。

返回:

Dict,key为参数名,value为该参数的切片策略。

异常:
  • ValueError - 策略文件不正确。

  • TypeError - strategy_filename 不是str。

样例:

>>> import mindspore as ms
>>> strategy = ms.build_searched_strategy("./strategy_train.ckpt")