Browse Source

inference.ipynb: reverting fp16 inference for now

master
rafaelvalle 6 years ago
parent
commit
825ffa47d1
1 changed files with 2 additions and 4 deletions
  1. +2
    -4
      inference.ipynb

+ 2
- 4
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)"
]
}

Loading…
Cancel
Save