Disable extra losses

This commit is contained in:
Eren Gölge 2022-02-22 13:54:53 +01:00
parent 89dd89b5e5
commit 4f68ba7127
1 changed files with 6 additions and 0 deletions

View File

@ -39,6 +39,12 @@ config = Tacotron2Config( # This is the config that is saved for the future use
run_eval=True,
test_delay_epochs=-1,
ga_alpha=0.0,
decoder_loss_alpha=0.25,
postnet_loss_alpha=0.25,
postnet_diff_spec_alpha=0,
decoder_diff_spec_alpha=0,
decoder_ssim_alpha=0,
postnet_ssim_alpha=0,
r=2,
attention_type="dynamic_convolution",
double_decoder_consistency=False,