Update modeling_super_linear.py
Browse files- modeling_super_linear.py +1 -1
modeling_super_linear.py
CHANGED
|
@@ -232,7 +232,7 @@ class RLinear(nn.Module):
|
|
| 232 |
print(W_3d.shape)
|
| 233 |
W_pool = F.adaptive_avg_pool1d(W_3d, output_size=new_lookback)
|
| 234 |
print(W_pool.shape)
|
| 235 |
-
W_new = W_pool.squeeze(0).
|
| 236 |
print(W_new.shape)
|
| 237 |
self.zero_shot_Linear = W_new
|
| 238 |
|
|
|
|
| 232 |
print(W_3d.shape)
|
| 233 |
W_pool = F.adaptive_avg_pool1d(W_3d, output_size=new_lookback)
|
| 234 |
print(W_pool.shape)
|
| 235 |
+
W_new = W_pool.squeeze(0).transpose(0, 1).contiguous()
|
| 236 |
print(W_new.shape)
|
| 237 |
self.zero_shot_Linear = W_new
|
| 238 |
|