razmars commited on
Commit
b34e6b8
·
verified ·
1 Parent(s): d3e03c1

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -213,7 +213,7 @@ class RLinear(nn.Module):
213
  fixed_weights = W[:, :L]
214
  dynamic_weights = W[:, L:]
215
 
216
- if in_features != self.weights.size(1):
217
  dynamic_weights = F.interpolate(dynamic_weights.unsqueeze(0).unsqueeze(0), size=(self.horizon, in_features-self.seq_len), mode='bilinear', align_corners=False).squeeze(0).squeeze(0)
218
  if self.fixed_in != 0:
219
  fixed_weights = F.interpolate(fixed_weights.unsqueeze(0).unsqueeze(0), size=(self.horizon, L), mode='bilinear', align_corners=False).squeeze(0).squeeze(0)
 
213
  fixed_weights = W[:, :L]
214
  dynamic_weights = W[:, L:]
215
 
216
+ if in_features != W.size(1):
217
  dynamic_weights = F.interpolate(dynamic_weights.unsqueeze(0).unsqueeze(0), size=(self.horizon, in_features-self.seq_len), mode='bilinear', align_corners=False).squeeze(0).squeeze(0)
218
  if self.fixed_in != 0:
219
  fixed_weights = F.interpolate(fixed_weights.unsqueeze(0).unsqueeze(0), size=(self.horizon, L), mode='bilinear', align_corners=False).squeeze(0).squeeze(0)