From f0d64126c649fb9879e83f68d14afbf82ad7d4ab Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@zalando.de>
Date: Sat, 4 Jul 2020 01:10:22 +0200
Subject: [PATCH] remove initialization

---
 pts/modules/distribution_output.py | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/pts/modules/distribution_output.py b/pts/modules/distribution_output.py
index c387e18..971d4ef 100644
--- a/pts/modules/distribution_output.py
+++ b/pts/modules/distribution_output.py
@@ -43,11 +43,6 @@ class ArgProj(nn.Module):
             [nn.Linear(in_features, dim) for dim in args_dim.values()]
         )
         self.domain_map = domain_map
-        
-        map(self._init_weights, self.proj)
-    
-    def _init_weights(self, m):
-        nn.init.kaiming_normal_(m.weight)
 
     def forward(self, x: torch.Tensor) -> Tuple[torch.Tensor]:
         params_unbounded = [proj(x) for proj in self.proj]