Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -231,7 +231,7 @@ class RLinear(nn.Module):
|
|
| 231 |
# resize H → self.horizon and W → new_lookback
|
| 232 |
new_W = F.interpolate(
|
| 233 |
W4d,
|
| 234 |
-
size=(self.seq_len
|
| 235 |
mode='bilinear',
|
| 236 |
align_corners=False
|
| 237 |
)[0, 0] # drop the two singleton dims
|
|
|
|
| 231 |
# resize H → self.horizon and W → new_lookback
|
| 232 |
new_W = F.interpolate(
|
| 233 |
W4d,
|
| 234 |
+
size=( new_lookback,self.seq_len), # (H_out, W_out)
|
| 235 |
mode='bilinear',
|
| 236 |
align_corners=False
|
| 237 |
)[0, 0] # drop the two singleton dims
|