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,11 +525,10 @@ 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, scale_factor=scale_factor, mode='linear')
+            x = interpolate(x_enc.permute(0, 2, 1), scale_factor=scale_factor, mode='bicubic')
 
-            #x = x[:,: , -self.seq_len:]
-            x = x[:,-self.seq_len:]
+
+            x = x[:,: , -self.seq_len:]
             orig_L = L
             L = self.seq_len