initialize weights

LLNL · Oct 23, 2024 · dc6736f · dc6736f
1 parent 0daa0b1
commit dc6736f
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]))