From 8dec2a9e95894891a72e98869185f5e39d78dd83 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 2 Oct 2019 00:30:25 +0200 Subject: [PATCH] fix memory leak duee to diagonal alingmnet score --- train.py | 4 ++-- utils/measures.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index b94f44fa..5ecf99ef 100644 --- a/train.py +++ b/train.py @@ -204,7 +204,7 @@ def train(model, criterion, criterion_st, optimizer, optimizer_st, scheduler, "GradNormST:{:.5f} AvgTextLen:{:.1f} AvgSpecLen:{:.1f} StepTime:{:.2f} " "LoaderTime:{:.2f} LR:{:.6f}".format( num_iter, batch_n_iter, global_step, - postnet_loss.item(), decoder_loss.item(), stop_loss.item(), align_score.item(), + postnet_loss.item(), decoder_loss.item(), stop_loss.item(), align_score, grad_norm, grad_norm_st, avg_text_length, avg_spec_length, step_time, loader_time, current_lr), flush=True) @@ -404,7 +404,7 @@ def evaluate(model, criterion, criterion_st, ap, global_step, epoch): postnet_loss.item(), keep_avg['avg_postnet_loss'], decoder_loss.item(), keep_avg['avg_decoder_loss'], stop_loss.item(), keep_avg['avg_stop_loss'], - align_score.item(), keep_avg['avg_align_score']), + align_score, keep_avg['avg_align_score']), flush=True) if args.rank == 0: diff --git a/utils/measures.py b/utils/measures.py index 21b61298..a76a2225 100644 --- a/utils/measures.py +++ b/utils/measures.py @@ -8,4 +8,4 @@ def alignment_diagonal_score(alignments): Shape: alignments : batch x decoder_steps x encoder_steps """ - return alignments.max(dim=1)[0].mean(dim=1).mean(dim=0) + return alignments.max(dim=1)[0].mean(dim=1).mean(dim=0).item()