20 lines
1.2 KiB
ReStructuredText
20 lines
1.2 KiB
ReStructuredText
mindspore.parallel.parameter_broadcast
|
||
============================================================================
|
||
|
||
.. py:function:: mindspore.parallel.parameter_broadcast(net, layout, cur_rank=0, initial_rank=0)
|
||
|
||
在数据并行维度，将参数广播到其他卡上。
|
||
|
||
.. warning::
|
||
实验性接口，未来可能变更或移除。
|
||
|
||
参数：
|
||
- **net** (Cell) - 将进行参数广播的网络。
|
||
- **layout** (Dict) - 参数layout字典，来自于函数 :func:`mindspore.nn.Cell.parameter_layout_dict` ；也可以从从文件中读取。例如，通过 :func:`mindspore.parallel.auto_parallel.AutoParallel.save_param_strategy_file` 保存下来的strategy.ckpt。该字典的key是参数名称，value是参数的Layout。
|
||
- **cur_rank** (int，可选) - 当前的rankID。默认值： ``0`` 。
|
||
- **initial_rank** (int，可选) - 每个流水线并行阶段的起始rankID。默认值： ``0`` 。
|
||
|
||
异常：
|
||
- **ValueError** - `cur_rank` 不是当前rank的rankID。
|
||
- **ValueError** - `initial_rank` 不是当前流水线并行阶段的起始rankID。
|
||
- **ValueError** - `layout` 中的参数名称不在函数 :func:`mindspore.nn.Cell.parameters_dict` 。 |