Update modeling_super_linear.py

modeling_super_linear.py

@@ -227,13 +227,8 @@ class RLinear(nn.Module):
             self.zero_shot_Linear = new_W                  # shape (self.horizon, new_lookback)
         else:
             W = self.Linear.weight.detach()
-            print(W.shape)
-            W_3d   = W.unsqueeze(0)   
-            print(W_3d.shape)
-            W_new  = F.adaptive_avg_pool1d(W_3d, new_lookback).squeeze(0)
-            print(W_new.shape)
-            self.zero_shot_Linear =  W_new
-            
+            m      = nn.AdaptiveAvgPool1d(new_lookback)
+            self.zero_shot_Linear = m(W)
         
 
     def forward(self, x):