wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-10 11:50:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

bug fix vocoder model restore

Browse Source
This commit is contained in:
erogol
2020-06-15 19:04:05 +02:00
parent b736c6cdac
commit 1daba4371d
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vocoder/configs/multiband-melgan_and_rwd_config.json
+2 -2
View File
@@ -56,8 +56,8 @@
"stft_loss_weight": 0.5,
"subband_stft_loss_weight": 0.5,
"mse_gan_loss_weight": 2.5,
"hinge_gan_loss_weight": 2.5,
"mse_G_loss_weight": 2.5,
"hinge_G_loss_weight": 2.5,
"feat_match_loss_weight": 25.0,
"stft_loss_params": {
vocoder/train.py
+1 -1
View File
@@ -421,7 +421,7 @@ def main(args): # pylint: disable=redefined-outer-name
optimizer_gen.load_state_dict(checkpoint['optimizer'])
model_disc.load_state_dict(checkpoint['model_disc'])
optimizer_disc.load_state_dict(checkpoint['optimizer_disc'])
except KeyError:
except RuntimeError:
print(" > Partial model initialization.")
model_dict = model_gen.state_dict()
model_dict = set_init_dict(model_dict, checkpoint['model'], c)