From 0bb2e5352e9cef57f0639d28fe9c2a483cb8d37f Mon Sep 17 00:00:00 2001
From: Kevin Chung <dreamer2368@gmail.com>
Date: Thu, 10 Oct 2024 16:39:20 -0700
Subject: [PATCH] initialize weights

---
 src/lasdi/latent_space.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/lasdi/latent_space.py b/src/lasdi/latent_space.py
index b1ad363..fdf8fef 100644
--- 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]))