From ec13243ff46aa490a3d7ffe4b16da874a9ebf649 Mon Sep 17 00:00:00 2001 From: TianYuan Date: Thu, 20 Apr 2023 08:58:16 +0000 Subject: [PATCH] add docstring for Discriminator --- paddlespeech/t2s/models/starganv2_vc/starganv2_vc.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/paddlespeech/t2s/models/starganv2_vc/starganv2_vc.py b/paddlespeech/t2s/models/starganv2_vc/starganv2_vc.py index 236708906..99aeb73bf 100644 --- 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