From bcda3a2c82ee4ba4d00c375cc7c89eacf3a5a2b9 Mon Sep 17 00:00:00 2001 From: Alex Reichenbach <8087473+Reichenbachian@users.noreply.github.com> Date: Wed, 16 Apr 2025 10:11:37 -0700 Subject: [PATCH] asdf --- swift/trainers/rlhf_trainer/reward_trainer.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/swift/trainers/rlhf_trainer/reward_trainer.py b/swift/trainers/rlhf_trainer/reward_trainer.py index 0355343909..069d0b0cc5 100644 --- a/swift/trainers/rlhf_trainer/reward_trainer.py +++ b/swift/trainers/rlhf_trainer/reward_trainer.py @@ -76,3 +76,9 @@ def visualize_samples(self, num_print_samples: int): if wandb.run is not None: wandb.log({'completions': wandb.Table(dataframe=df)}) + elif 'neptune' in self.args.report_to: + import neptune + from neptune.types import File + + if neptune.run is not None: + neptune.run['completions'].upload(File.as_html(df))