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

train.py - add with torch.no_grad():

This commit is contained in:
Eren Golge 2018-05-10 16:44:37 -07:00
parent 7c40455edd
commit b4bd713581
1 changed files with 33 additions and 32 deletions

1
train.py
View File

@ -191,6 +191,7 @@ def evaluate(model, criterion, data_loader, current_step):
print(" | > Validation") print(" | > Validation")
progbar = Progbar(len(data_loader.dataset) / c.batch_size) progbar = Progbar(len(data_loader.dataset) / c.batch_size)
n_priority_freq = int(3000 / (c.sample_rate * 0.5) * c.num_freq) n_priority_freq = int(3000 / (c.sample_rate * 0.5) * c.num_freq)
with torch.no_grad():
for num_iter, data in enumerate(data_loader): for num_iter, data in enumerate(data_loader):
start_time = time.time() start_time = time.time()