Update modeling_super_linear.py
Browse files- 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=(
|
| 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
|