diff --git a/trl/trainer/reward_trainer.py b/trl/trainer/reward_trainer.py index 3b98d68d90..c3fc904fa6 100644 --- a/trl/trainer/reward_trainer.py +++ b/trl/trainer/reward_trainer.py @@ -310,7 +310,6 @@ def visualize_samples(self, num_print_samples: int): if num_print_samples >= 0 and len(table["chosen_text"]) >= num_print_samples: break df = pd.DataFrame(table) - print_rich_table(pd.DataFrame(table)) if self.accelerator.process_index == 0: print_rich_table(df[:num_print_samples]) if "wandb" in self.args.report_to: