From e5c349f0b019cbc54dd9eaace98a5f2cdc0f7f8b Mon Sep 17 00:00:00 2001
From: pranz24 <pranjal.tandon@gmail.com>
Date: Sat, 6 Jun 2020 09:34:03 +0530
Subject: [PATCH] tensorboard cleanup

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index e6ac07b..85973de 100644
--- a/main.py
+++ b/main.py
@@ -56,8 +56,8 @@ env.seed(args.seed)
 # Agent
 agent = SAC(env.observation_space.shape[0], env.action_space, args)
 
-#TesnorboardX
-writer = SummaryWriter(logdir='runs/{}_SAC_{}_{}_{}'.format(datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S"), args.env_name,
+#Tesnorboard
+writer = SummaryWriter('runs/{}_SAC_{}_{}_{}'.format(datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S"), args.env_name,
                                                              args.policy, "autotune" if args.automatic_entropy_tuning else ""))
 
 # Memory