diff --git a/tft/estimator.py b/tft/estimator.py index c4ac35d..9086c38 100644 --- a/tft/estimator.py +++ b/tft/estimator.py @@ -268,7 +268,6 @@ class TFTEstimator(PyTorchLightningEstimator): input_names=PREDICTION_INPUT_NAMES, prediction_net=module.model, batch_size=self.batch_size, - freq=self.freq, prediction_length=self.prediction_length, device=torch.device("cuda" if torch.cuda.is_available() else "cpu"), )