razmars commited on
Commit
53ee76f
·
verified ·
1 Parent(s): 4341f95

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +2 -2
modeling_super_linear.py CHANGED
@@ -223,11 +223,10 @@ class RLinear(nn.Module):
223
  align_corners=False
224
  )[0, 0] # drop the two singleton dims
225
 
226
- print(new_W.shape)
227
-
228
  self.zero_shot_Linear = new_W # shape (self.horizon, new_lookback)
229
 
230
 
 
231
  def forward(self, x):
232
  # x: [Batch, Input length,Channel]
233
  x_shape = x.shape
@@ -236,6 +235,7 @@ class RLinear(nn.Module):
236
  #print(F"new Lookkback : {x.shape[1]}")
237
 
238
  self.transform_model(x.shape[1],3)
 
239
  x = x.clone()
240
  #x = x * (x.shape[1]/512)
241
  x = self.revin_layer(x, 'norm')
 
223
  align_corners=False
224
  )[0, 0] # drop the two singleton dims
225
 
 
 
226
  self.zero_shot_Linear = new_W # shape (self.horizon, new_lookback)
227
 
228
 
229
+
230
  def forward(self, x):
231
  # x: [Batch, Input length,Channel]
232
  x_shape = x.shape
 
235
  #print(F"new Lookkback : {x.shape[1]}")
236
 
237
  self.transform_model(x.shape[1],3)
238
+
239
  x = x.clone()
240
  #x = x * (x.shape[1]/512)
241
  x = self.revin_layer(x, 'norm')