razmars commited on
Commit
203f7df
·
verified ·
1 Parent(s): 399c990

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -235,7 +235,7 @@ class RLinear(nn.Module):
235
 
236
  new_W = F.interpolate(
237
  W4d,
238
- size=(out_features, self.seq_len), # keep rows = out_features,
239
  # stretch cols to seq_len
240
  mode='bilinear',
241
  align_corners=False
 
235
 
236
  new_W = F.interpolate(
237
  W4d,
238
+ size=(new_lookback, self.seq_len), # keep rows = out_features,
239
  # stretch cols to seq_len
240
  mode='bilinear',
241
  align_corners=False