diff --git a/README.md b/README.md index ce89b09..8fd3d04 100644 --- a/README.md +++ b/README.md @@ -60,7 +60,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") estimator = DeepAREstimator(freq="5min", prediction_length=12, - input_size=43, + input_size=19, trainer=Trainer(epochs=10, device=device)) predictor = estimator.train(training_data=training_data, num_workers=4)