From 0a7b9e51fbc9e004e5f5288fa9dd0bd21ea687bf Mon Sep 17 00:00:00 2001
From: Pranjal Tandon <pranjal.tandon@gmail.com>
Date: Wed, 2 Jan 2019 22:24:04 +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 6b2724f..5f21717 100644
--- a/model.py
+++ b/model.py
@@ -13,7 +13,7 @@ epsilon = 1e-6
 def weights_init(m):
     classname = m.__class__.__name__
     if classname.find('Linear') != -1:
-        torch.nn.init.xavier_normal_(m.weight)
+        torch.nn.init.xavier_uniform_(m.weight)
 
 
 class ValueNetwork(nn.Module):