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

ipynb typo

Browse Source
This commit is contained in:
Raul Puri
2018-05-05 17:30:08 -07:00
committed by GitHub
parent c67ca6531e
commit 4ac6ce9ab5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
inference.ipynb
View File

@@ -101,7 +101,7 @@
"try:\n", "try:\n",
" model = model.module\n", " model = model.module\n",
"except:\n", "except:\n",
" pass\n" " pass\n",
"model.load_state_dict({k.replace('module.',''):v for k,v in torch.load(checkpoint_path)['state_dict'].items()})\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()"
] ]