Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -492,7 +492,7 @@ class superLinear(nn.Module):
|
|
| 492 |
return cycle
|
| 493 |
|
| 494 |
|
| 495 |
-
def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=
|
| 496 |
if len(x_enc.shape) > 2:
|
| 497 |
x = x_enc.permute(0, 2, 1)
|
| 498 |
B, V, L = x.shape
|
|
|
|
| 492 |
return cycle
|
| 493 |
|
| 494 |
|
| 495 |
+
def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
|
| 496 |
if len(x_enc.shape) > 2:
|
| 497 |
x = x_enc.permute(0, 2, 1)
|
| 498 |
B, V, L = x.shape
|