We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fe928a9 commit 3461b55Copy full SHA for 3461b55
swift/rlhf_trainers/utils.py
@@ -593,7 +593,7 @@ def profiling_context(trainer, name: str):
593
is_main_process = trainer.is_main_process
594
595
if 'wandb' in trainer.args.report_to and wandb.run is not None and is_main_process:
596
- wandb.log(profiling_metrics)
+ wandb.log(profiling_metrics, commit=False)
597
598
if 'swanlab' in trainer.args.report_to and swanlab.get_run() is not None and is_main_process:
599
swanlab.log(profiling_metrics)
0 commit comments