fix: weight init

Signed-off-by: Avik Basu <ab93@users.noreply.github.com>
numaproj · Jun 12, 2024 · 09f2c01 · 09f2c01
1 parent b775e1c
commit 09f2c01
Show file tree

Hide file tree

Showing 2 changed files with 76 additions and 82 deletions.
diff --git a/examples/vanilla_ic.ipynb b/examples/vanilla_ic.ipynb
diff --git a/numalogic/models/autoencoder/variants/icvanilla.py b/numalogic/models/autoencoder/variants/icvanilla.py
@@ -178,12 +178,6 @@ def __init__(
         self.encoder.apply(self.init_weights)
         self.decoder.apply(self.init_weights)
 
-    @staticmethod
-    def init_weights(m: nn.Module) -> None:
-        """Initialize the parameters in the model."""
-        if isinstance(m, nn.Linear):
-            nn.init.xavier_normal_(m.weight)
-
     def forward(self, batch: Tensor) -> tuple[Tensor, Tensor]:
         batch = torch.swapdims(batch, 1, 2)
         encoded = self.encoder(batch)