razmars commited on
Commit
0134478
·
verified ·
1 Parent(s): 9627b54

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -1
modeling_super_linear.py CHANGED
@@ -228,7 +228,7 @@ class RLinear(nn.Module):
228
  else:
229
  W = self.Linear.weight.detach()
230
  print(W.shape)
231
- W_3d = W.t().unsqueeze(0)
232
  print(W_3d.shape)
233
  W_pool = F.adaptive_avg_pool1d(W_3d, output_size=new_lookback)
234
  print(W_pool.shape)
 
228
  else:
229
  W = self.Linear.weight.detach()
230
  print(W.shape)
231
+ W_3d = W.unsqueeze(0)
232
  print(W_3d.shape)
233
  W_pool = F.adaptive_avg_pool1d(W_3d, output_size=new_lookback)
234
  print(W_pool.shape)