From 6440e47a0b27df2dc15932145dbcb7bbaa756282 Mon Sep 17 00:00:00 2001
From: Pranjal Tandon <pranjal.tandon@gmail.com>
Date: Wed, 5 Dec 2018 02:28:37 +0530
Subject: [PATCH] Update model.py

---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model.py b/model.py
index 6bb6ccc..9b965d5 100644
--- a/model.py
+++ b/model.py
@@ -121,7 +121,7 @@ class DeterministicPolicy(nn.Module):
         x = inputs
         x = F.relu(self.linear1(x))
         x = F.relu(self.linear2(x))
-        mean = F.tanh(self.mu(x))
+        mean = F.tanh(self.mean(x))
         return mean