Update modeling_super_linear.py
Browse files- 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 |
|