Skip to content

Commit 861e065

Browse files
authored
Update ctc_loss function parameters in documentation
Added parameters 'norm_by_times' and 'zero_infinity' to ctc_loss function documentation.
1 parent 60beaad commit 861e065

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

docs/api/paddle/nn/functional/ctc_loss_cn.rst

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
ctc_loss
44
-------------------------------
55

6-
.. py:function:: paddle.nn.functional.ctc_loss(log_probs, labels, input_lengths, label_lengths, blank=0, reduction='mean')
6+
.. py:function:: paddle.nn.functional.ctc_loss(log_probs, labels, input_lengths, label_lengths, blank=0, reduction='mean', norm_by_times=False, zero_infinity=False)
77
88
计算 CTC loss。该接口的底层调用了第三方 baidu-research::warp-ctc 的实现。
99
也可以叫做 softmax with CTC,因为 Warp-CTC 库中插入了 softmax 激活函数来对输入的值进行归一化。
@@ -17,6 +17,7 @@ ctc_loss
1717
- **blank** (int,可选) - 空格标记的 ID 值,其取值范围为 [0, num_classes + 1)。数据类型支持 int32。默认值为 0。
1818
- **reduction** (str,可选) - 指定应用于输出结果的计算方式,可选值有:``'none'``, ``'mean'``, ``'sum'``。设置为 ``'mean'`` 时,对 loss 值除以 label_lengths,并返回所得商的均值;设置为 ``'sum'`` 时,返回 loss 值的总和;设置为 ``'none'`` 时,则直接返回输出的 loss 值。默认值为 ``'mean'``。
1919
- **norm_by_times** (bool,可选) - 是否根据序列长度对梯度进行正则化。数据类型支持 bool。默认值为 False。
20+
- **zero_infinity** (bool,可选) - 防止梯度出现无限,将其重置为0,以免训练时发生爆炸。
2021

2122
返回
2223
:::::::::

0 commit comments

Comments
 (0)