From 7a3f5d29617a2ddf1b8dafc7356ad9615ef05873 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Wed, 23 Dec 2020 12:42:57 +0100 Subject: [PATCH] fix api --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 70904af..45edfa2 100644 --- a/README.md +++ b/README.md @@ -63,7 +63,7 @@ estimator = DeepAREstimator(freq="5min", input_size=43, trainer=Trainer(epochs=10, device=device)) -predictor = estimator.train(training_data=training_data, num_workers=4, num_prefetch=2) +predictor = estimator.train(training_data=training_data, num_workers=4) ``` ``` 45it [00:01, 37.60it/s, avg_epoch_loss=4.64, epoch=0]