Commit 17bf6050 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

reverted to tensorboardX due to HPC demands

parent a4fc6fde
......@@ -16,7 +16,7 @@ setup(
'torch',
'h5py',
'fslpy',
'tensorboard',
'tensorboardX',
'sklearn',
],
)
......@@ -25,7 +25,7 @@ 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 torch.utils.tensorboard import SummaryWriter
from tensorboardX import SummaryWriter
import utils.data_evaluation_utils as evaluation
......@@ -64,8 +64,8 @@ class LogWriter():
shutil.rmtree(validation_logs_directory)
self.log_writer = {
'train': SummaryWriter(log_dir=training_logs_directory),
'validation': SummaryWriter(log_dir=validation_logs_directory)
'train': SummaryWriter(logdir=training_logs_directory),
'validation': SummaryWriter(logdir=validation_logs_directory)
}
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