razmars commited on
Commit
25a4f7f
·
verified ·
1 Parent(s): d78c6b4

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. 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 , new_lookback), # (H_out, W_out)
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