Update modeling_super_linear.py

modeling_super_linear.py

@@ -212,7 +212,7 @@ class RLinear(nn.Module):
             new_W          = new_W * final_scaling
 
             self.zero_shot_Linear        = new_W
-        else:
+        elif mode ==2:
             W = self.Linear.weight.detach()
             W4d = W.unsqueeze(0).unsqueeze(0)            # (1, 1, out, in)
 
@@ -225,6 +225,13 @@ class RLinear(nn.Module):
                     )[0, 0]                                 # drop the two singleton dims
 
             self.zero_shot_Linear = new_W                  # shape (self.horizon, new_lookback)
+        else:
+            W = self.Linear.weight.detach()
+            W_3d   = W.t().unsqueeze(0)                     
+            W_pool = F.adaptive_avg_pool1d(W_3d, output_size=new_lookback)   
+            W_new  = W_pool.squeeze(0).permute(2, 0).contiguous()
+            self.zero_shot_Linear =  
+            
         
 
     def forward(self, x):
@@ -234,7 +241,7 @@ class RLinear(nn.Module):
             #if self.zero_shot_Linear is None:
                 #print(F"new Lookkback : {x.shape[1]}")
             
-            self.transform_model(x.shape[1],1)
+            self.transform_model(x.shape[1],3)
 
             x = x.clone()
             #x = x * (x.shape[1]/512)