fix api change

This commit is contained in:
Kashif Rasul
2022-06-22 12:14:35 +02:00
parent ab0c038700
commit 289f2ba3e0
-1
View File
@@ -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"),
)