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

changed test logs name to validation

parent 4c63df04
...@@ -53,19 +53,19 @@ class LogWriter(): ...@@ -53,19 +53,19 @@ class LogWriter():
self.number_of_classes = number_of_classes self.number_of_classes = number_of_classes
training_logs_directory = os.path.join( training_logs_directory = os.path.join(
logs_directory, experiment_name, "train") logs_directory, experiment_name, "train")
testing_logs_directory = os.path.join( validation_logs_directory = os.path.join(
logs_directory, experiment_name, "test") logs_directory, experiment_name, "validation")
# If the logs directory exist, we clear their contents to allow new logs to be created # If the logs directory exist, we clear their contents to allow new logs to be created
if not use_last_checkpoint: if not use_last_checkpoint:
if os.path.exists(training_logs_directory): if os.path.exists(training_logs_directory):
shutil.rmtree(training_logs_directory) shutil.rmtree(training_logs_directory)
if os.path.exists(testing_logs_directory): if os.path.exists(validation_logs_directory):
shutil.rmtree(testing_logs_directory) shutil.rmtree(validation_logs_directory)
self.log_writer = { self.log_writer = {
'train': SummaryWriter(log_dir=training_logs_directory), 'train': SummaryWriter(log_dir=training_logs_directory),
'test': SummaryWriter(log_dir=testing_logs_directory) 'validation': SummaryWriter(log_dir=validation_logs_directory)
} }
self.confusion_matrix_color_map = confusion_matrix_cmap self.confusion_matrix_color_map = confusion_matrix_cmap
...@@ -247,4 +247,4 @@ class LogWriter(): ...@@ -247,4 +247,4 @@ class LogWriter():
""" """
self.log_writer['train'].close() self.log_writer['train'].close()
self.log_writer['test'].close() self.log_writer['validation'].close()
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