do not merge: hack scripting to get it to work

DataWaveProject · Feb 5, 2025 · c389d75 · c389d75
1 parent 67cad7c
commit c389d75
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/utils/model_definition.py b/utils/model_definition.py
@@ -71,11 +71,11 @@ def __init__(self, idim, odim, hdim, stencil, dropout=0.0):
         self.output = nn.Linear(2 * odim, odim)
 
     def forward(self, x):
-        if self.fac == 1:
-            x = torch.squeeze(self.dropout0(self.act_cnn(self.conv1(x))))
-        elif self.fac == 2:
-            x = torch.squeeze(self.dropout0(self.act_cnn(self.conv1(x))))
-            x = torch.squeeze(self.dropout0_2(self.act_cnn2(self.conv2(x))))
+        # if self.fac == 1:
+        #     x = torch.squeeze(self.dropout0(self.act_cnn(self.conv1(x))))
+        # elif self.fac == 2:
+        #     x = torch.squeeze(self.dropout0(self.act_cnn(self.conv1(x))))
+        #     x = torch.squeeze(self.dropout0_2(self.act_cnn2(self.conv2(x))))
 
         # print(f'new shape: {x.shape}')
         x = self.dropout(self.act1(self.layer1(x)))