malarinv
/
tacotron2
mirror of https://github.com/malarinv/tacotron2
1
0
Fork 0
Code Issues Releases Wiki Activity

inference.ipynb: reverting fp16 inference for now

experiments
rafaelvalle 2018-12-08 21:26:01 -08:00
parent 4d7b04120a
commit 825ffa47d1
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
inference.ipynb
View File

@ -116,9 +116,7 @@
"source": [
"waveglow_path = 'waveglow_old.pt'\n",
"waveglow = torch.load(waveglow_path)['model']\n",
"waveglow.cuda().half()\n",
"for k in waveglow.convinv:\n",
" k.float()"
"waveglow.cuda()"
]
},
{
@ -205,7 +203,7 @@
],
"source": [
"with torch.no_grad():\n",
" audio = waveglow.infer(mel_outputs_postnet.half(), sigma=0.666)\n",
" audio = waveglow.infer(mel_outputs_postnet, sigma=0.666)\n",
"ipd.Audio(audio[0].data.cpu().numpy(), rate=hparams.sampling_rate)"
]
}