remove initialization

2026-07-17 11:32:26 +08:00 · 2020-07-04 01:10:22 +02:00
parent ead4609f91
commit f0d64126c6
1 changed files with 0 additions and 5 deletions
@@ -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]