Skip to content

Commit 3461b55

Browse files
committed
fix wandb log
1 parent fe928a9 commit 3461b55

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/rlhf_trainers/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -593,7 +593,7 @@ def profiling_context(trainer, name: str):
593593
is_main_process = trainer.is_main_process
594594

595595
if 'wandb' in trainer.args.report_to and wandb.run is not None and is_main_process:
596-
wandb.log(profiling_metrics)
596+
wandb.log(profiling_metrics, commit=False)
597597

598598
if 'swanlab' in trainer.args.report_to and swanlab.get_run() is not None and is_main_process:
599599
swanlab.log(profiling_metrics)

0 commit comments

Comments
 (0)