From 301c705c4301972a0edf1595830929d9a8d0626e Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Sun, 20 Jul 2025 20:12:50 +0200 Subject: [PATCH] fix(reforcexy): fix total steps computation 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 | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py index ab09827..6893975 100644 --- a/ReforceXY/user_data/freqaimodels/ReforceXY.py +++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py @@ -330,17 +330,18 @@ class ReforceXY(BaseReinforcementLearningModel): test_df = data_dictionary.get("test_features") test_timesteps = len(test_df) train_cycles = max(1, int(self.rl_config.get("train_cycles", 25))) - total_timesteps = train_timesteps * train_cycles * self.n_envs + total_timesteps = ( + (train_timesteps * train_cycles + self.n_envs - 1) // self.n_envs + ) * self.n_envs train_days = steps_to_days(train_timesteps, self.config.get("timeframe")) total_days = steps_to_days(total_timesteps, self.config.get("timeframe")) logger.info("Action masking: %s", self.is_maskable) logger.info( - "Train: %s steps (%s days) * %s cycles * %s environments = Total %s (%s days)", + "Train: %s steps (%s days) * %s cycles = Total %s (%s days)", train_timesteps, train_days, train_cycles, - self.n_envs, total_timesteps, total_days, ) -- 2.43.0