Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -490,7 +490,7 @@ class superLinear(nn.Module):
|
|
| 490 |
return cycle
|
| 491 |
|
| 492 |
|
| 493 |
-
def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
|
| 494 |
|
| 495 |
if inf_pred_len is None:
|
| 496 |
inf_pred_len = self.inf_pred_len
|
|
|
|
| 490 |
return cycle
|
| 491 |
|
| 492 |
|
| 493 |
+
def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False, inf_pred_len=None):
|
| 494 |
|
| 495 |
if inf_pred_len is None:
|
| 496 |
inf_pred_len = self.inf_pred_len
|