diff --git a/utils/summaries.py b/utils/summaries.py index f714e6ac..76aa8e95 100644 --- a/utils/summaries.py +++ b/utils/summaries.py @@ -9,7 +9,7 @@ def __init__(self, directory): self.directory = directory def create_summary(self): - writer = SummaryWriter(log_dir=os.path.join(self.directory)) + writer = SummaryWriter(logdir=os.path.join(self.directory)) return writer def visualize_image(self, writer, dataset, image, target, output, global_step):