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

cache empty

This commit is contained in:
Eren Gölge 2021-04-08 11:12:21 +02:00
parent 3fb78c004a
commit 8daf407652
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TTS/bin/train_vocoder_gan.py
View File

@ -306,6 +306,7 @@ def train(model_G, criterion_G, optimizer_G, model_D, criterion_D, optimizer_D,
# TODO: plot model stats # TODO: plot model stats
# if c.tb_model_param_stats: # if c.tb_model_param_stats:
# tb_logger.tb_model_weights(model, global_step) # tb_logger.tb_model_weights(model, global_step)
torch.cuda.empty_cache()
return keep_avg.avg_values, global_step return keep_avg.avg_values, global_step
@ -433,9 +434,6 @@ def evaluate(model_G, criterion_G, model_D, criterion_D, ap, global_step, epoch)
if c.print_eval: if c.print_eval:
c_logger.print_eval_step(num_iter, loss_dict, keep_avg.avg_values) c_logger.print_eval_step(num_iter, loss_dict, keep_avg.avg_values)
torch.cuda.empty_cache()
if args.rank == 0: if args.rank == 0:
# compute spectrograms # compute spectrograms
figures = plot_results(y_hat, y_G, ap, global_step, 'eval') figures = plot_results(y_hat, y_G, ap, global_step, 'eval')
@ -450,7 +448,7 @@ def evaluate(model_G, criterion_G, model_D, criterion_D, ap, global_step, epoch)
# synthesize a full voice # synthesize a full voice
data_loader.return_segments = False data_loader.return_segments = False
torch.cuda.empty_cache()
return keep_avg.avg_values return keep_avg.avg_values