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