diff --git a/logger.py b/logger.py index e6422c5..ad32738 100644 --- a/logger.py +++ b/logger.py @@ -1,6 +1,6 @@ import random import torch -from tensorboardX import SummaryWriter +from torch.utils.tensorboard import SummaryWriter from plotting_utils import plot_alignment_to_numpy, plot_spectrogram_to_numpy from plotting_utils import plot_gate_outputs_to_numpy diff --git a/requirements.txt b/requirements.txt index 11eccea..660bde9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,6 +4,5 @@ numpy==1.13.3 inflect==0.2.5 librosa==0.6.0 scipy==1.0.0 -tensorboardX==1.1 Unidecode==1.0.22 pillow