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 bd68492e41371f5f870f5b960fe05287420dc8ba..0f40489e8d12e4564b616483bb2d5f5eb030db05 100644 (file)
--- 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,