razmars commited on
Commit
7a86a83
·
verified ·
1 Parent(s): b672046

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +1 -0
modeling_super_linear.py CHANGED
@@ -196,6 +196,7 @@ class RLinear(nn.Module):
196
  super(RLinear, self).__init__()
197
  self.Linear = nn.Linear(input_len, output_len)
198
  self.seq_len = input_len
 
199
  self.revin_layer = RevIN(num_features = None, affine=False, norm_type = None, subtract_last = False)
200
  self.zero_shot_Linear = None
201
 
 
196
  super(RLinear, self).__init__()
197
  self.Linear = nn.Linear(input_len, output_len)
198
  self.seq_len = input_len
199
+ self.horizon = output_len
200
  self.revin_layer = RevIN(num_features = None, affine=False, norm_type = None, subtract_last = False)
201
  self.zero_shot_Linear = None
202