From cb8057a58367c52a474a5382ccc2338262c8724d Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Tue, 18 Feb 2025 17:42:08 +0100 Subject: [PATCH] refactor(reforcexy): cleanups MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- ReforceXY/user_data/freqaimodels/ReforceXY.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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, -- 2.43.0