Update modeling_super_linear.py

modeling_super_linear.py

@@ -226,10 +226,14 @@ class RLinear(nn.Module):
 
             self.zero_shot_Linear = new_W                  # shape (self.horizon, new_lookback)
         else:
-            W = self.Linear.weight.detach()
-            W_3d   = W.t().unsqueeze(0)                     
+            W = self.Linear.weight.detach
+            print(W.shape)
+            W_3d   = W.t().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).permute(2, 0).contiguous()
+            print(W_new.shape)
             self.zero_shot_Linear =  W_new