Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

small fix

This commit is contained in:
Eren Golge 2019-12-09 13:27:46 +01:00
parent 956f8b7672
commit 40a533530b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -592,7 +592,7 @@ def main(args): # pylint: disable=redefined-outer-name
args.restore_step = 0 args.restore_step = 0
if use_cuda: if use_cuda:
model = model.cuda() model.cuda()
criterion.cuda() criterion.cuda()
if criterion_st: if criterion_st:
criterion_st.cuda() criterion_st.cuda()