Skip to content

Commit d6175c6

Browse files
author
Jan Beitner
committed
Improve base_model docs
1 parent c7b3933 commit d6175c6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytorch_forecasting/models/base_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -159,8 +159,8 @@ def __init__(self, my_first_parameter: int=2, loss=SMAPE()):
159159
super().__init__(loss=loss)
160160
161161
def forward(self, x):
162-
encoding_target = x["encoder_target"]
163-
prediction = self.transform_output(prediction=..., target_scale=x["target_scale"])
162+
normalized_prediction = self.module(x)
163+
prediction = self.transform_output(prediction=normalized_prediction, target_scale=x["target_scale"])
164164
return self.to_network_output(prediction=prediction)
165165
166166
"""

0 commit comments

Comments
 (0)