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)