razmars commited on
Commit
9cced13
·
verified ·
1 Parent(s): 0134478

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. 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).permute(2, 0).contiguous()
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