Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -0
modeling_super_linear.py
CHANGED
|
@@ -211,6 +211,7 @@ class RLinear(nn.Module):
|
|
| 211 |
|
| 212 |
self.zero_shot_Linear = new_W
|
| 213 |
else:
|
|
|
|
| 214 |
W4d = W.unsqueeze(0).unsqueeze(0) # (1, 1, out, in)
|
| 215 |
|
| 216 |
# resize H → self.horizon and W → new_lookback
|
|
|
|
| 211 |
|
| 212 |
self.zero_shot_Linear = new_W
|
| 213 |
else:
|
| 214 |
+
W = self.Linear.weight.detach()
|
| 215 |
W4d = W.unsqueeze(0).unsqueeze(0) # (1, 1, out, in)
|
| 216 |
|
| 217 |
# resize H → self.horizon and W → new_lookback
|