fix(ReforceXY): exclude gpu_memory_fraction from model params

diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py
index b62f7071c5536bfc8774449bd0cf11c196641961..d003a3a6c680d441493a9fac79d5fbe8377fb08f 100644 (file)
--- a/ReforceXY/user_data/freqaimodels/ReforceXY.py
+++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py
@@ -760,6 +760,8 @@ class ReforceXY(BaseReinforcementLearningModel):
             )  # "adamw"
         )
 
+        model_params.pop("gpu_memory_fraction", None)
+
         self._model_params_cache = model_params
         return copy.deepcopy(self._model_params_cache)