razmars commited on
Commit
25f53ff
·
verified ·
1 Parent(s): a7dc532

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. 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