From 691d325c14679f19a85b2b779093d393fae91408 Mon Sep 17 00:00:00 2001 From: Yichen Zhou Date: Fri, 10 May 2024 11:28:01 -0700 Subject: [PATCH] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 40c759f..b7e5fb7 100644 --- a/README.md +++ b/README.md @@ -94,6 +94,8 @@ model_dims=1280, 2. The horizon length can be set to anything. We recommend setting it to the largest horizon length you would need in the forecasting tasks for your application. We generally recommend horizon length <= context length but it is not a requirement in the function call. +3. `backend` is one of "cpu", "gpu" or "tpu", case sensitive. + ### Perform inference We provide APIs to forecast from either array inputs or `pandas` dataframe. Both forecast methods expect (1) the input time series contexts, (2) along with their frequencies. Please look at the documentation of the functions `tfm.forecast()` and `tfm.forecast_on_df()` for detailed instructions.