From dbbbacc39dcc8979503094f4edd7561525ceb601 Mon Sep 17 00:00:00 2001
From: pranz24 <pranjal.tandon@gmail.com>
Date: Sat, 6 Jun 2020 08:57:18 +0530
Subject: [PATCH] remove tensorboardX

---
 README.md | 1 -
 main.py   | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/README.md b/README.md
index f9267e3..59401e1 100644
--- a/README.md
+++ b/README.md
@@ -9,7 +9,6 @@ Learning with a Stochastic Actor](https://arxiv.org/pdf/1801.01290.pdf) -> [SAC_
 ### Requirements
 ------------
 *   [mujoco-py](https://github.com/openai/mujoco-py)
-*   [TensorboardX](https://github.com/lanpa/tensorboardX)
 *   [PyTorch](http://pytorch.org/)
 
 ### Default Arguments and Usage
diff --git a/main.py b/main.py
index a3047c5..e6ac07b 100644
--- a/main.py
+++ b/main.py
@@ -5,7 +5,7 @@ import numpy as np
 import itertools
 import torch
 from sac import SAC
-from tensorboardX import SummaryWriter
+from torch.utils.tensorboard import SummaryWriter
 from replay_memory import ReplayMemory
 
 parser = argparse.ArgumentParser(description='PyTorch Soft Actor-Critic Args')