close fig to avoid memory leak

Kyubyong · Mar 27, 2018 · 531bd03 · 531bd03
1 parent 6375051
commit 531bd03
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils.py b/utils.py
@@ -129,6 +129,7 @@ def plot_alignment(alignment, gs, dir=hp.logdir):
     fig.colorbar(im)
     plt.title('{} Steps'.format(gs))
     plt.savefig('{}/alignment_{}.png'.format(dir, gs), format='png')
+    plt.close(fig)
 
 def guided_attention(g=0.2):
     '''Guided attention. Refer to page 3 on the paper.'''