mindspore.build_searched_strategy
- class mindspore.build_searched_strategy(strategy_filename)[源代码]
构建网络中每个参数的策略,用于分布式推理。关于它的使用细节,请参考: 保存和加载模型(HyBrid Parallel模式)。
参数:
strategy_filename (str) - 策略文件的名称。
返回:
Dict,key为参数名,value为该参数的切片策略。
异常:
ValueError - 策略文件不正确。
TypeError - strategy_filename 不是str。
样例:
>>> strategy = build_searched_strategy("./strategy_train.ckpt")