lirannoc commited on
Commit
fc20137
·
verified ·
1 Parent(s): 167a566

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. 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