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

prompt data loade time per iteartion

pull/10/head
Eren Golge 2019-07-22 15:10:19 +02:00
parent 1827f77752
commit 713b3df792
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
train.py
View File

@ -180,6 +180,9 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
else: else:
grad_norm_st = 0 grad_norm_st = 0
step_time = time.time() - start_time
epoch_time += step_time
if current_step % c.print_step == 0: if current_step % c.print_step == 0:
print( print(
" | > Step:{}/{} GlobalStep:{} TotalLoss:{:.5f} PostnetLoss:{:.5f} " " | > Step:{}/{} GlobalStep:{} TotalLoss:{:.5f} PostnetLoss:{:.5f} "
@ -242,9 +245,7 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
tb_logger.tb_train_audios(current_step, tb_logger.tb_train_audios(current_step,
{'TrainAudio': train_audio}, {'TrainAudio': train_audio},
c.audio["sample_rate"]) c.audio["sample_rate"])
end_time = time.time()
step_time = end_time - start_time
epoch_time += step_time
avg_postnet_loss /= (num_iter + 1) avg_postnet_loss /= (num_iter + 1)
avg_decoder_loss /= (num_iter + 1) avg_decoder_loss /= (num_iter + 1)
@ -274,8 +275,6 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler,
tb_logger.tb_train_epoch_stats(current_step, epoch_stats) tb_logger.tb_train_epoch_stats(current_step, epoch_stats)
if c.tb_model_param_stats: if c.tb_model_param_stats:
tb_logger.tb_model_weights(model, current_step) tb_logger.tb_model_weights(model, current_step)
end_time = time.time()
return avg_postnet_loss, current_step return avg_postnet_loss, current_step