Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -0
modeling_super_linear.py
CHANGED
|
@@ -392,6 +392,7 @@ class Model(nn.Module):
|
|
| 392 |
- Prediction tensor
|
| 393 |
- (Optional) Expert selection probabilities if get_prob is True
|
| 394 |
"""
|
|
|
|
| 395 |
if pred_len is None:
|
| 396 |
pred_len = self.train_pred_len
|
| 397 |
|
|
|
|
| 392 |
- Prediction tensor
|
| 393 |
- (Optional) Expert selection probabilities if get_prob is True
|
| 394 |
"""
|
| 395 |
+
print(pred_len)
|
| 396 |
if pred_len is None:
|
| 397 |
pred_len = self.train_pred_len
|
| 398 |
|