Update modeling_super_linear.py

modeling_super_linear.py

@@ -512,7 +512,7 @@ class superLinear(nn.Module):
             B, V, L = x.shape
         else:
             x     = x_enc
-            x_enc =  x_enc.unsqueeze(-1)
+            #x_enc =  x_enc.unsqueeze(-1)
             B, L = x.shape
             V    = 1
 
@@ -525,9 +525,11 @@ class superLinear(nn.Module):
             orig_pred_len = inf_pred_len
 
             inf_pred_len = inf_pred_len*scale_factor
+            #x = interpolate(x_enc.permute(0, 2, 1), scale_factor=scale_factor, mode='linear')
             x = interpolate(x_enc.permute(0, 2, 1), scale_factor=scale_factor, mode='linear')
 
-            x = x[:,: , -self.seq_len:]
+            #x = x[:,: , -self.seq_len:]
+            x = x[:,-self.seq_len:]
             orig_L = L
             L = self.seq_len