changed logging util from tensorboardX to torch.utils.tensorboard

setup.py

@@ -15,7 +15,6 @@ setup(
         'pandas',
         'torch',
         'h5py',
-        'tensorboardX',
         'fslpy'
         ],
 )

utils/data_logging_utils.py

@@ -24,7 +24,8 @@ import torch
 
 # The SummaryWriter class provides a high-level API to create an event file in a given directory and add summaries and events to it.
 # More here: https://tensorboardx.readthedocs.io/en/latest/tensorboard.html
-from tensorboardX import SummaryWriter
+
+from torch.utils.tensorboard import SummaryWriter
 
 import utils.data_evaluation_utils as evaluation
 
@@ -63,8 +64,8 @@ class LogWriter():
                 shutil.rmtree(testing_logs_directory)
 
         self.log_writer = {
-            'train': SummaryWriter(logdir=training_logs_directory),
-            'test': SummaryWriter(logdir=testing_logs_directory)
+            'train': SummaryWriter(log_dir=training_logs_directory),
+            'test': SummaryWriter(log_dir=testing_logs_directory)
         }
 
         self.confusion_matrix_color_map = confusion_matrix_cmap