From 3057d4d8dd7744300b9ea8369728e5ac6308332b Mon Sep 17 00:00:00 2001 From: Zach Zhu Date: Thu, 14 Dec 2023 16:07:43 +0800 Subject: [PATCH] algo: make tsf use a reasonable small batch size as default Signed-off-by: Zach Zhu --- algorithm/kapacity/timeseries/forecasting/forecaster.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/algorithm/kapacity/timeseries/forecasting/forecaster.py b/algorithm/kapacity/timeseries/forecasting/forecaster.py index 0c7b6d2..d986ed9 100644 --- a/algorithm/kapacity/timeseries/forecasting/forecaster.py +++ b/algorithm/kapacity/timeseries/forecasting/forecaster.py @@ -630,7 +630,7 @@ def fit(freq: str, context_length: int, learning_rate: float = 1e-3, epochs: int = 100, - batch_size: int = 1024, + batch_size: int = 32, num_workers: int = 0, model_path: str = './', df: pd.DataFrame = None,