add docstring for Discriminator

2 years ago · ec13243ff4
parent dee6630945
commit ec13243ff4
1 changed files with 10 additions and 0 deletions
--- a/paddlespeech/t2s/models/starganv2_vc/starganv2_vc.py
+++ b/paddlespeech/t2s/models/starganv2_vc/starganv2_vc.py
@ -555,6 +555,16 @@ class Discriminator(nn.Layer):
        self.reset_parameters()

    def forward(self, x: paddle.Tensor, y: paddle.Tensor):
+        """Calculate forward propagation.
+        Args:
+            x(Tensor(float32)):
+                Shape (B, 1, 80, T).
+            y(Tensor(float32)):
+                Shape (B, ). 
+        Returns:
+            Tensor:
+                Shape (B, )
+        """
        out = self.dis(x, y)
        return out