changing model to four inputs

deepskies · Nov 8, 2023 · 32a976a · 32a976a
1 parent b8ce7bb
commit 32a976a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/scripts/models.py b/src/scripts/models.py
@@ -13,7 +13,7 @@ class de_no_var(nn.Module):
     def __init__(self):
         super().__init__()
         drop_percent = 0.1
-        self.ln_1 = nn.Linear(3, 100)
+        self.ln_1 = nn.Linear(4, 100)
         self.act1 = nn.ReLU()
         self.drop1 = nn.Dropout(drop_percent)
         self.ln_2 = nn.Linear(100, 100)
@@ -36,7 +36,7 @@ class de_var(nn.Module):
     def __init__(self):
         super().__init__()
         drop_percent = 0.1
-        self.ln_1 = nn.Linear(3, 100)
+        self.ln_1 = nn.Linear(4, 100)
         self.act1 = nn.ReLU()
         self.drop1 = nn.Dropout(drop_percent)
         self.ln_2 = nn.Linear(100, 100)