From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Wed, 2 Apr 2025 18:49:08 +0000 (+0200)
Subject: perf(qav3): limit optuna memory usage
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=50af9f15f17bb0e9d794cf9d524d4b3ae67cb100;p=freqai-strategies.git

perf(qav3): limit optuna memory usage

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---

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