Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2025-12-25 20:59:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

cache empty

Browse Source
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
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
# if c.tb_model_param_stats:
# tb_logger.tb_model_weights(model, global_step)
torch.cuda.empty_cache()
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:
c_logger.print_eval_step(num_iter, loss_dict, keep_avg.avg_values)
torch.cuda.empty_cache()
if args.rank == 0:
# compute spectrograms
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
data_loader.return_segments = False
torch.cuda.empty_cache()
return keep_avg.avg_values