Update modeling_super_linear.py

modeling_super_linear.py

@@ -204,8 +204,8 @@ class RLinear(nn.Module):
     def transform_model(self,new_lookback,mode):
         if mode == 1:
             W              = self.Linear.weight.detach()
-            #new_W          = W[:, -new_lookback:] 
-            new_W          = W[:, :new_lookback]
+            new_W          = W[:, -new_lookback:] 
+            #new_W          = W[:, :new_lookback]
             original_norm  = torch.norm(W, p=2)
             new_norm       = torch.norm(new_W, p=2)
             final_scaling  = original_norm / new_norm if new_norm.item() != 0 else 1.0
@@ -227,7 +227,6 @@ class RLinear(nn.Module):
             self.zero_shot_Linear = new_W                  # shape (self.horizon, new_lookback)
         
 
-
     def forward(self, x):
         # x: [Batch, Input length,Channel]
         x_shape = x.shape