enables training the decoder, fixes matplotlib multi-threading issue #2160 (#2174)

b14765c3 · Oier Mees · Reed · 58fb6d7e · b14765c3
Commit b14765c3 authored Aug 17, 2017 by Oier Mees Committed by Reed Aug 16, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

ptn/utils.py ptn/utils.py +2 -0

No files found.
--- a/ptn/utils.py
+++ b/ptn/utils.py
@@ -20,6 +20,8 @@ from __future__ import division
 from __future__ import print_function

 import StringIO
+import matplotlib
+matplotlib.use('Agg')
 from matplotlib import pylab as p
 # axes3d is being used implictly for visualization.
 from mpl_toolkits.mplot3d import axes3d as p3  # pylint:disable=unused-import