Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -510,7 +510,7 @@ class superLinear(nn.Module):
|
|
| 510 |
return cycle
|
| 511 |
|
| 512 |
|
| 513 |
-
def forward(self, x_enc, x_mark_enc=None, x_dec=None, x_mark_dec=None, mask=None, freq=[None], get_prob=False):
|
| 514 |
|
| 515 |
if inf_pred_len is None:
|
| 516 |
inf_pred_len = self.inf_pred_len
|
|
|
|
| 510 |
return cycle
|
| 511 |
|
| 512 |
|
| 513 |
+
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):
|
| 514 |
|
| 515 |
if inf_pred_len is None:
|
| 516 |
inf_pred_len = self.inf_pred_len
|