From b523701867972b39d8010af0df6a05458feef3b6 Mon Sep 17 00:00:00 2001 From: TianYuan Date: Thu, 20 Apr 2023 09:03:56 +0000 Subject: [PATCH] add typehint --- paddlespeech/t2s/models/starganv2_vc/losses.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/paddlespeech/t2s/models/starganv2_vc/losses.py b/paddlespeech/t2s/models/starganv2_vc/losses.py index f9ff39276..aef7559f9 100644 --- a/paddlespeech/t2s/models/starganv2_vc/losses.py +++ b/paddlespeech/t2s/models/starganv2_vc/losses.py @@ -27,9 +27,9 @@ 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.): @@ -37,7 +37,6 @@ def compute_d_loss(nets: Dict[str, Any], 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)