add typehint

PaddlePaddle · Apr 20, 2023 · b523701 · b523701
1 parent ec13243
commit b523701
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/paddlespeech/t2s/models/starganv2_vc/losses.py b/paddlespeech/t2s/models/starganv2_vc/losses.py
@@ -27,17 +27,16 @@ def compute_d_loss(nets: Dict[str, Any],
                    y_trg: paddle.Tensor,
                    z_trg: paddle.Tensor=None,
                    x_ref: paddle.Tensor=None,
-                   use_r1_reg=True,
-                   use_adv_cls=False,
-                   use_con_reg=False,
+                   use_r1_reg: bool=True,
+                   use_adv_cls: bool=False,
+                   use_con_reg: bool=False,
                    lambda_reg: float=1.,
                    lambda_adv_cls: float=0.1,
                    lambda_con_reg: float=10.):
 
     assert (z_trg is None) != (x_ref is None)
     # with real audios
     x_real.stop_gradient = False
-
     out = nets['discriminator'](x_real, y_org)
     loss_real = adv_loss(out, 1)