From 81be06bcc128729ad8901fcf1c722834f176ac34 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Tue, 2 Aug 2022 09:32:58 -0400 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)