From 616ca73a68b41411f2057bcde5bc96c214f22fb2 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Thu, 16 Jan 2020 15:17:37 +0100 Subject: [PATCH] comment out graph for now and close it at the end --- pts/trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pts/trainer.py b/pts/trainer.py index 394df73..0c95795 100644 --- a/pts/trainer.py +++ b/pts/trainer.py @@ -37,9 +37,7 @@ class Trainer: ) writer = SummaryWriter() - - writer.add_graph(net) - + #writer.add_graph(net) for epoch_no in range(self.epochs): # mark epoch start time @@ -78,3 +76,5 @@ class Trainer: # mark epoch end time and log time cost of current epoch toc = time.time() + + writer.close()