increase memory_limit to 32 GB

limpbot · Nov 11, 2022 · 910997a · 910997a
1 parent 71c373d
commit 910997a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frameworks/AutoPyTorch/exec.py b/frameworks/AutoPyTorch/exec.py
@@ -87,7 +87,7 @@ def run(dataset, config):
             X_test=X_test,
             optimize_metric=eval_metric,
             n_prediction_steps=forecast_horizon,
-            memory_limit=16 * 1024,  # Currently, forecasting models use much more memories
+            memory_limit=32 * 1024,  # Currently, forecasting models use much more memories
             freq=freq, #FREQ_MAP[freq],
             start_times=start_times,
             #func_eval_time_limit_secs=50,