algo: make tsf use a reasonable small batch size as default

Signed-off-by: Zach Zhu <[email protected]>
traas-stack · Dec 14, 2023 · 3057d4d · 3057d4d
1 parent 1364020
commit 3057d4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,