From 50af9f15f17bb0e9d794cf9d524d4b3ae67cb100 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 2 Apr 2025 20:49:08 +0200 Subject: [PATCH] perf(qav3): limit optuna memory usage MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/QuickAdapterRegressorV3.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py index f0394cf..8b0087b 100644 --- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py +++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py @@ -319,7 +319,7 @@ class QuickAdapterRegressorV3(BaseRegressionModel): params_storage: dict[str, dict], rmse_storage: dict[str, float], ) -> None: - identifier = self.freqai_info["identifier"] + identifier = self.freqai_info.get("identifier") study = self.optuna_create_study(f"{identifier}-{namespace}-{pair}", pair) if not study: return @@ -332,9 +332,10 @@ class QuickAdapterRegressorV3(BaseRegressionModel): try: study.optimize( objective, - n_trials=self.__optuna_config["n_trials"], - n_jobs=self.__optuna_config["n_jobs"], - timeout=self.__optuna_config["timeout"], + n_trials=self.__optuna_config.get("n_trials"), + n_jobs=self.__optuna_config.get("n_jobs"), + timeout=self.__optuna_config.get("timeout"), + gc_after_trial=True, ) except Exception as e: time_spent = time.time() - start_time -- 2.43.0