Update MCTEM.py

Author: LRQ577

models/MCTEM.py

@@ -77,11 +77,11 @@ def __init__(self, configs):
         # @lrq 映射输出维度
         self.to_output = nn.Linear(self.seq_len, self.pred_len)       
 
-        self.fc = nn.Sequential(
-            nn.Linear(self.seq_length * self.embed_size, self.hidden_size),
-            nn.LeakyReLU(),
-            nn.Linear(self.hidden_size, self.pre_length)
-        )
+        # self.fc = nn.Sequential(
+        #     nn.Linear(self.seq_length * self.embed_size, self.hidden_size),
+        #     nn.LeakyReLU(),
+        #     nn.Linear(self.hidden_size, self.pre_length)
+        # )
         # @lrq合并用的函数
         self.W_fuse = torch.nn.Parameter(torch.ones(2))
 
@@ -214,4 +214,4 @@ def forward(self, x):
         x = self.relu1(self.layer1(x))
         x = self.relu2(self.layer2(x))
         x = self.layer3(x)
-        return x
+        return x