initialize weights

LLNL · Oct 10, 2024 · 0bb2e53 · 0bb2e53
1 parent 2381d61
commit 0bb2e53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/lasdi/latent_space.py b/src/lasdi/latent_space.py
@@ -66,6 +66,7 @@ def __init__(self, layer_sizes,
         for k in range(self.n_layers-1):
             self.fcs += [torch.nn.Linear(layer_sizes[k], layer_sizes[k + 1])]
         self.fcs = torch.nn.ModuleList(self.fcs)
+        self.init_weight()
 
         # Reshape input or output layer
         assert((reshape_index is None) or (reshape_index in [0, -1]))