malarinv/tacotron2
1
0
Fork 0
mirror of https://github.com/malarinv/tacotron2 synced 2026-03-08 01:32:35 +00:00
Code Issues Releases Wiki Activity

Merge branch 'master' of https://github.com/NVIDIA/tacotron2

Browse Source
This commit is contained in:
Rafael Valle
2018-05-06 08:58:07 -07:00
parent dcd925f6c8 4ac6ce9ab5
commit 20568643cb
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
inference.ipynb
View File

@@ -98,8 +98,11 @@
"source": [ "source": [
"checkpoint_path = \"/home/scratch.adlr-gcf/audio_denoising/runs/TTS-Tacotron2-LJS-MSE-DRC-NoMaskPadding-Unsorted-Distributed-22khz/checkpoint_15500\"\n", "checkpoint_path = \"/home/scratch.adlr-gcf/audio_denoising/runs/TTS-Tacotron2-LJS-MSE-DRC-NoMaskPadding-Unsorted-Distributed-22khz/checkpoint_15500\"\n",
"model = load_model(hparams)\n", "model = load_model(hparams)\n",
"model.load_state_dict(torch.load(checkpoint_path)['state_dict'])\n", "try:\n",
"model = model.module\n", " model = model.module\n",
"except:\n",
" pass\n",
"model.load_state_dict({k.replace('module.',''):v for k,v in torch.load(checkpoint_path)['state_dict'].items()})\n",
"_ = model.eval()" "_ = model.eval()"
] ]
}, },