razmars commited on
Commit
0310ef6
·
verified ·
1 Parent(s): 27c6437

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +2 -2
modeling_super_linear.py CHANGED
@@ -239,11 +239,11 @@ class RLinear(nn.Module):
239
  #print(F"new Lookkback : {x.shape[1]}")
240
 
241
  self.transform_model(x.shape[1],1)
242
- #x = x * (x.shape[1]/512)
243
  x = self.revin_layer(x, 'norm')
244
  x = F.linear(x, self.zero_shot_Linear)
245
  x = self.revin_layer(x, 'denorm')
246
- #x = x * (512/x.shape[1])
247
  return x
248
 
249
 
 
239
  #print(F"new Lookkback : {x.shape[1]}")
240
 
241
  self.transform_model(x.shape[1],1)
242
+ x = x * (x.shape[1]/512)
243
  x = self.revin_layer(x, 'norm')
244
  x = F.linear(x, self.zero_shot_Linear)
245
  x = self.revin_layer(x, 'denorm')
246
+ x = x * (512/x.shape[1])
247
  return x
248
 
249