Update modeling_super_linear.py

modeling_super_linear.py

@@ -452,11 +452,13 @@ class superLinear(nn.Module):
             x = x_enc.permute(0, 2, 1)
             B, V, L = x.shape
         else:
-            x = x_enc
+            x    = x_enc
             B, L = x.shape
             V    = 1
             
         x = x.reshape(B * V, L)
+        print("RAZ")
+        print(x.shape)
         
         expert_probs = None
         
@@ -473,7 +475,9 @@ class superLinear(nn.Module):
                 outputs.append(ar_out)
                 ar_x = torch.cat([ar_x, ar_out], dim=1)[:, -self.seq_len:]
             out = torch.cat(outputs, dim=1)[:, :self.inf_pred_len]
+        print(F"out1 :{out.shape}")
         out = out.reshape(B, V, out.shape[-1])
+        print(F"out2 :{out.shape}")
         result = out.permute(0, 2, 1)
         
         if get_prob: