From 32a976a563cd7b1094e3c2bdb42eb4c423eb1b88 Mon Sep 17 00:00:00 2001 From: beckynevin Date: Wed, 8 Nov 2023 16:40:57 -0700 Subject: [PATCH] changing model to four inputs --- src/scripts/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/scripts/models.py b/src/scripts/models.py index ae0ab04..acf8c4f 100644 --- 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)