Fixed missing import and using x11

+ Added `time` import in `train_tacotron.py` * Modified `utils/display.py` to only use x11 when necessary. This will allow training the model on servers where x11 is either not installed or not working (e.g. due to improperly forwarded X11)
scarlettlin2022 · Jul 26, 2019 · 1ed4591 · 1ed4591
1 parent 1a947d0
commit 1ed4591
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/train_tacotron.py b/train_tacotron.py
@@ -11,6 +11,7 @@
 from utils import data_parallel_workaround
 import os
 from pathlib import Path
+import time
 
 
 def np_now(x: torch.Tensor): return x.detach().cpu().numpy()

diff --git a/utils/display.py b/utils/display.py
@@ -1,3 +1,5 @@
+import matplotlib as mpl
+mpl.use('agg')  # Use non-interactive backend by default
 import matplotlib.pyplot as plt
 import time
 import numpy as np
@@ -96,6 +98,7 @@ def save_spectrogram(M, path, length=None):
 
 
 def plot(array): 
+    mpl.interactive(True)
     fig = plt.figure(figsize=(30, 5))
     ax = fig.add_subplot(111)
     ax.xaxis.label.set_color('grey')
@@ -105,11 +108,14 @@ def plot(array):
     ax.tick_params(axis='x', colors='grey', labelsize=23)
     ax.tick_params(axis='y', colors='grey', labelsize=23)
     plt.plot(array)
+    mpl.interactive(False)
 
 
 def plot_spec(M):
+    mpl.interactive(True)
     M = np.flip(M, axis=0)
     plt.figure(figsize=(18,4))
     plt.imshow(M, interpolation='nearest', aspect='auto')
     plt.show()
+    mpl.interactive(False)