Update AutoGluon max_memory from 0.1 to 0.4 in persist_models (#543)

openml · Jun 21, 2023 · bb25b01 · bb25b01
1 parent aeb709d
commit bb25b01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frameworks/AutoGluon/exec.py b/frameworks/AutoGluon/exec.py
@@ -68,7 +68,7 @@ def run(dataset, config):
         )
 
     # Persist model in memory that is going to be predicting to get correct inference latency
-    predictor.persist_models('best')
+    predictor.persist_models('best', max_memory=0.4)
 
     def inference_time_classification(data: Union[str, pd.DataFrame]):
         return None, predictor.predict_proba(data, as_multiclass=True)