From 0cea873fdf6ddbf0897d03259b23767953184f29 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Sun, 17 Aug 2025 22:33:11 +0200 Subject: [PATCH] fix(reforcexy): DQN properties for tensorbard 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 | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index 996069e..24a6558 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -1363,11 +1363,12 @@ class InfoMetricsCallback(TensorboardCallback): { "buffer_size": self.model.buffer_size, "gradient_steps": self.model.gradient_steps, - "train_freq": self.model.train_freq, "learning_starts": self.model.learning_starts, "target_update_interval": self.model.target_update_interval, "exploration_fraction": self.model.exploration_fraction, + "exploration_initial_eps": self.model.exploration_initial_eps, "exploration_final_eps": self.model.exploration_final_eps, + "exploration_rate": self.model.exploration_rate, } ) if "QRDQN" in self.model.__class__.__name__: -- 2.43.0