Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -525,7 +525,7 @@ class superLinear(nn.Module):
|
|
| 525 |
orig_pred_len = inf_pred_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='
|
| 529 |
|
| 530 |
|
| 531 |
x = x[:,: , -self.seq_len:]
|
|
|
|
| 525 |
orig_pred_len = inf_pred_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 |
|
| 530 |
|
| 531 |
x = x[:,: , -self.seq_len:]
|