dan
/
tacotron2
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

ipynb typo

master
Raul Puri 6 years ago
committed by GitHub
parent
c67ca6531e
commit
4ac6ce9ab5
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      inference.ipynb

+ 1
- 1
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()"
] ]

Write Preview
Loading…
Cancel
Save