logger.py: compatibility with new tensorboardX

NVIDIA · Dec 12, 2019 · 37a033d · 37a033d
1 parent 70d37f9
commit 37a033d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/logger.py b/logger.py
@@ -31,18 +31,18 @@ def log_validation(self, reduced_loss, model, y, y_pred, iteration):
         self.add_image(
             "alignment",
             plot_alignment_to_numpy(alignments[idx].data.cpu().numpy().T),
-            iteration)
+            iteration, dataformats='HWC')
         self.add_image(
             "mel_target",
             plot_spectrogram_to_numpy(mel_targets[idx].data.cpu().numpy()),
-            iteration)
+            iteration, dataformats='HWC')
         self.add_image(
             "mel_predicted",
             plot_spectrogram_to_numpy(mel_outputs[idx].data.cpu().numpy()),
-            iteration)
+            iteration, dataformats='HWC')
         self.add_image(
             "gate",
             plot_gate_outputs_to_numpy(
                 gate_targets[idx].data.cpu().numpy(),
                 torch.sigmoid(gate_outputs[idx]).data.cpu().numpy()),
-            iteration)
+            iteration, dataformats='HWC')