razmars commited on
Commit
5b56ca0
·
verified ·
1 Parent(s): d82ef44

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -526,7 +526,7 @@ class superLinear(nn.Module):
526
 
527
  inf_pred_len = inf_pred_len*scale_factor
528
  #x = interpolate(x_enc.permute(0, 2, 1), scale_factor=scale_factor, mode='linear')
529
- x = interpolate(x_enc.permute(0, 2, 1), scale_factor=scale_factor, mode='linear')
530
 
531
  #x = x[:,: , -self.seq_len:]
532
  x = x[:,-self.seq_len:]
 
526
 
527
  inf_pred_len = inf_pred_len*scale_factor
528
  #x = interpolate(x_enc.permute(0, 2, 1), scale_factor=scale_factor, mode='linear')
529
+ x = interpolate(x_enc, scale_factor=scale_factor, mode='linear')
530
 
531
  #x = x[:,: , -self.seq_len:]
532
  x = x[:,-self.seq_len:]