!30991 modify content order in rst file for mastet part2
Merge pull request !30991 from lvmingfu/master
This commit is contained in:
commit
382d3959f0
|
@ -15,11 +15,11 @@ mindspore.nn.transformer
|
||||||
.. include:: transformer/mindspore.nn.Transformer.rst
|
.. include:: transformer/mindspore.nn.Transformer.rst
|
||||||
.. include:: transformer/mindspore.nn.TransformerOpParallelConfig.rst
|
.. include:: transformer/mindspore.nn.TransformerOpParallelConfig.rst
|
||||||
.. include:: transformer/mindspore.nn.EmbeddingOpParallelConfig.rst
|
.. include:: transformer/mindspore.nn.EmbeddingOpParallelConfig.rst
|
||||||
|
.. include:: transformer/mindspore.nn.TransformerRecomputeConfig.rst
|
||||||
.. include:: transformer/mindspore.nn.CrossEntropyLoss.rst
|
.. include:: transformer/mindspore.nn.CrossEntropyLoss.rst
|
||||||
.. include:: transformer/mindspore.nn.OpParallelConfig.rst
|
.. include:: transformer/mindspore.nn.OpParallelConfig.rst
|
||||||
.. include:: transformer/mindspore.nn.FixedSparseAttention.rst
|
.. include:: transformer/mindspore.nn.FixedSparseAttention.rst
|
||||||
.. include:: transformer/mindspore.nn.MoEConfig.rst
|
.. include:: transformer/mindspore.nn.MoEConfig.rst
|
||||||
.. include:: transformer/mindspore.nn.TransformerRecomputeConfig.rst
|
|
||||||
|
|
||||||
.. automodule:: mindspore.nn.transformer
|
.. automodule:: mindspore.nn.transformer
|
||||||
:members:
|
:members:
|
||||||
|
|
|
@ -3,43 +3,6 @@
|
||||||
|
|
||||||
并行相关功能的接口。
|
并行相关功能的接口。
|
||||||
|
|
||||||
.. py:function:: get_algo_parameters(attr_key)
|
|
||||||
|
|
||||||
获取算法参数配置属性。
|
|
||||||
|
|
||||||
.. note::
|
|
||||||
属性名称为必填项。此接口仅在AUTO_PARALLEL模式下工作。
|
|
||||||
|
|
||||||
**参数:**
|
|
||||||
|
|
||||||
- **attr_key** (str) - 属性的key。key包括"fully_use_devices"、"elementwise_op_strategy_follow"、"enable_algo_approxi"、"algo_approxi_epsilon"、"tensor_slice_align_enable”和"tensor_slice_align_size"。
|
|
||||||
|
|
||||||
**返回:**
|
|
||||||
|
|
||||||
根据key返回属性值。
|
|
||||||
|
|
||||||
**异常:**
|
|
||||||
|
|
||||||
ValueError:无法识别传入的关键字。
|
|
||||||
|
|
||||||
|
|
||||||
.. py:function:: reset_algo_parameters()
|
|
||||||
|
|
||||||
重置算法参数属性。
|
|
||||||
|
|
||||||
.. note::
|
|
||||||
此接口仅在AUTO_PARALLEL模式下工作。
|
|
||||||
|
|
||||||
重置后,属性值为:
|
|
||||||
|
|
||||||
- fully_use_devices:True
|
|
||||||
- elementwise_op_strategy_follow:False
|
|
||||||
- enable_algo_approxi:False
|
|
||||||
- algo_approxi_epsilon:0.1
|
|
||||||
- tensor_slice_align_enable:False
|
|
||||||
- tensor_slice_align_size:16
|
|
||||||
|
|
||||||
|
|
||||||
.. py:function:: set_algo_parameters(**kwargs)
|
.. py:function:: set_algo_parameters(**kwargs)
|
||||||
|
|
||||||
设置并行策略搜索算法中的参数。有关典型用法,请参见mindspore/tests/ut/python/parallel/test_auto_parallel_resnet.py。
|
设置并行策略搜索算法中的参数。有关典型用法,请参见mindspore/tests/ut/python/parallel/test_auto_parallel_resnet.py。
|
||||||
|
@ -59,3 +22,40 @@
|
||||||
**异常:**
|
**异常:**
|
||||||
|
|
||||||
- **ValueError** - 无法识别传入的关键字。
|
- **ValueError** - 无法识别传入的关键字。
|
||||||
|
|
||||||
|
|
||||||
|
.. py:function:: reset_algo_parameters()
|
||||||
|
|
||||||
|
重置算法参数属性。
|
||||||
|
|
||||||
|
.. note::
|
||||||
|
此接口仅在AUTO_PARALLEL模式下工作。
|
||||||
|
|
||||||
|
重置后,属性值为:
|
||||||
|
|
||||||
|
- fully_use_devices:True
|
||||||
|
- elementwise_op_strategy_follow:False
|
||||||
|
- enable_algo_approxi:False
|
||||||
|
- algo_approxi_epsilon:0.1
|
||||||
|
- tensor_slice_align_enable:False
|
||||||
|
- tensor_slice_align_size:16
|
||||||
|
|
||||||
|
|
||||||
|
.. py:function:: get_algo_parameters(attr_key)
|
||||||
|
|
||||||
|
获取算法参数配置属性。
|
||||||
|
|
||||||
|
.. note::
|
||||||
|
属性名称为必填项。此接口仅在AUTO_PARALLEL模式下工作。
|
||||||
|
|
||||||
|
**参数:**
|
||||||
|
|
||||||
|
- **attr_key** (str) - 属性的key。key包括"fully_use_devices"、"elementwise_op_strategy_follow"、"enable_algo_approxi"、"algo_approxi_epsilon"、"tensor_slice_align_enable”和"tensor_slice_align_size"。
|
||||||
|
|
||||||
|
**返回:**
|
||||||
|
|
||||||
|
根据key返回属性值。
|
||||||
|
|
||||||
|
**异常:**
|
||||||
|
|
||||||
|
ValueError:无法识别传入的关键字。
|
||||||
|
|
Loading…
Reference in New Issue