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": [ "source": [
"waveglow_path = 'waveglow_old.pt'\n", "waveglow_path = 'waveglow_old.pt'\n",
"waveglow = torch.load(waveglow_path)['model']\n", "waveglow = torch.load(waveglow_path)['model']\n",
"waveglow.cuda().half()\n", "waveglow.cuda()"
"for k in waveglow.convinv:\n",
" k.float()"
] ]
}, },
{ {
@ -205,7 +203,7 @@
], ],
"source": [ "source": [
"with torch.no_grad():\n", "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)" "ipd.Audio(audio[0].data.cpu().numpy(), rate=hparams.sampling_rate)"
] ]
} }