diff --git a/configs/lightning_module/eben.yaml b/configs/lightning_module/eben.yaml index ac8bb80..f0f5a5d 100644 --- a/configs/lightning_module/eben.yaml +++ b/configs/lightning_module/eben.yaml @@ -4,7 +4,7 @@ sample_rate: ${sample_rate} dynamic_loss_balancing: "ema" beta_ema: 0.9 update_discriminator_ratio: 1 -description: "${description} || EBEN(M=4,P=${generator.p},Q=${discriminator.q})" +description: "${description} || EBEN(M=${generator.m},P=${generator.p},Q=${discriminator.q})" push_to_hub_after_testing: False defaults: