Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -0
modeling_super_linear.py
CHANGED
|
@@ -448,6 +448,7 @@ class superLinear(nn.Module):
|
|
| 448 |
|
| 449 |
|
| 450 |
def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
|
|
|
|
| 451 |
x = x_enc.permute(0, 2, 1)
|
| 452 |
B, V, L = x.shape
|
| 453 |
x = x.reshape(B * V, L)
|
|
|
|
| 448 |
|
| 449 |
|
| 450 |
def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
|
| 451 |
+
print(x_enc.shape)
|
| 452 |
x = x_enc.permute(0, 2, 1)
|
| 453 |
B, V, L = x.shape
|
| 454 |
x = x.reshape(B * V, L)
|