Commit 16550d2e authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

fixed bug in log_writer dictionary key 'test'

parent b3a42e70
...@@ -64,8 +64,8 @@ class LogWriter(): ...@@ -64,8 +64,8 @@ class LogWriter():
self.log_writer = { self.log_writer = {
'train': SummaryWriter(logdir=training_logs_directory), 'train': SummaryWriter(logdir=training_logs_directory),
'test:': SummaryWriter(logdir=testing_logs_directory) 'test': SummaryWriter(logdir=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