Skip to content

Commit 00febff

Browse files
committed
[vec][loss] update docstring, test=doc fix #1717
1 parent 1667577 commit 00febff

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddlespeech/vector/modules/loss.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,9 @@ def __init__(self, Q, noise_ratio=100, Z_offset=9.5):
132132

133133
def forward(self, output, target):
134134
"""Forward inference
135+
136+
Args:
137+
output (tensor): the model output, which is the input of loss function
135138
"""
136139
output = paddle.reshape(output, [-1, self.N])
137140
B = output.shape[0]

0 commit comments

Comments
 (0)