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

moved LogWriter initialization before load_checkpoint constructor

parent 4b4ac4d4
......@@ -108,15 +108,16 @@ class Solver():
self.best_mean_score = 0
self.best_mean_score_epoch = 0
if use_last_checkpoint:
self.load_checkpoint()
self.LogWriter = LogWriter(number_of_classes=number_of_classes,
logs_directory=logs_directory,
experiment_name=experiment_name,
use_last_checkpoint=use_last_checkpoint,
labels=labels)
if use_last_checkpoint:
self.load_checkpoint()
def train(self, train_loader, test_loader):
"""Training Function
......
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