From: Jérôme Benoit <jerome.benoit@sap.com>
Date: Tue, 18 Feb 2025 16:42:08 +0000 (+0100)
Subject: refactor(reforcexy): cleanups
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=cb8057a58367c52a474a5382ccc2338262c8724d;p=freqai-strategies.git

refactor(reforcexy): cleanups

Signed-off-by: Jérôme Benoit <jerome.benoit@sap.com>
---

diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py
index bd68492..0f40489 100644
--- a/ReforceXY/user_data/freqaimodels/ReforceXY.py
+++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py
@@ -1105,9 +1105,13 @@ class InfoMetricsCallback(TensorboardCallback):
 
     def _on_training_start(self) -> None:
         _lr = self.model.learning_rate
+        _lr = _lr if isinstance(_lr, float) else "lr_schedule"
+        _cr = self.model.clip_range
+        _cr = _cr if isinstance(_cr, float) else "cr_schedule"
         hparam_dict = {
             "algorithm": self.model.__class__.__name__,
-            "learning_rate": _lr if _lr is float else "lr_schedule",
+            "learning_rate": _lr,
+            "clip_range": _cr,
             # "gamma": self.model.gamma,
             # "gae_lambda": self.model.gae_lambda,
             # "n_steps": self.model.n_steps,