Commit 0b7bd09d authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

changed logging util from tensorboardX to torch.utils.tensorboard

parent 5514e9b2
...@@ -15,7 +15,6 @@ setup( ...@@ -15,7 +15,6 @@ setup(
'pandas', 'pandas',
'torch', 'torch',
'h5py', 'h5py',
'tensorboardX',
'fslpy' 'fslpy'
], ],
) )
...@@ -24,7 +24,8 @@ import torch ...@@ -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. # 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 # 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 import utils.data_evaluation_utils as evaluation
...@@ -63,8 +64,8 @@ class LogWriter(): ...@@ -63,8 +64,8 @@ class LogWriter():
shutil.rmtree(testing_logs_directory) shutil.rmtree(testing_logs_directory)
self.log_writer = { self.log_writer = {
'train': SummaryWriter(logdir=training_logs_directory), 'train': SummaryWriter(log_dir=training_logs_directory),
'test': SummaryWriter(logdir=testing_logs_directory) 'test': SummaryWriter(log_dir=testing_logs_directory)
} }
self.confusion_matrix_color_map = confusion_matrix_cmap self.confusion_matrix_color_map = confusion_matrix_cmap
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment