Add sb3 learn progress bar

2023-04-02 02:59:02 +03:00
parent dc7e834911
commit cab82e8e60
5 changed files with 9 additions and 2 deletions
--- a/freqtrade/freqai/prediction_models/ReinforcementLearner.py
+++ b/freqtrade/freqai/prediction_models/ReinforcementLearner.py
@@ -71,7 +71,8 @@ class ReinforcementLearner(BaseReinforcementLearningModel):

        model.learn(
            total_timesteps=int(total_timesteps),
-            callback=[self.eval_callback, self.tensorboard_callback]
+            callback=[self.eval_callback, self.tensorboard_callback],
+            progress_bar=self.freqai_info["rl_config"]["progress_bar"]
        )

        if Path(dk.data_path / "best_model.zip").is_file():