Update modeling_super_linear.py

modeling_super_linear.py

@@ -228,11 +228,9 @@ class RLinear(nn.Module):
         else:
             W = self.Linear.weight.detach()
             print(W.shape)
-            W_3d   = W.t().unsqueeze(0)  
+            W_3d   = W.unsqueeze(0)   
             print(W_3d.shape)
-            W_pool = F.adaptive_avg_pool1d(W_3d, output_size=new_lookback)   
-            print(W_pool.shape)
-            W_new  = W_pool.squeeze(0).transpose(0, 1).contiguous()
+            W_new  = F.adaptive_avg_pool1d(W_3d, new_lookback).squeeze(0)
             print(W_new.shape)
             self.zero_shot_Linear =  W_new