From 0dadf53674b9eefa108d76881e78b960f656ac96 Mon Sep 17 00:00:00 2001 From: Hank Kung Date: Mon, 17 Jun 2019 05:35:21 +0800 Subject: [PATCH] Fixed the typo fix the problem:TypeError: __init__() got an unexpected keyword argument 'log_dir' --- utils/summaries.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):