frozenleaves/DeepSpeed
1
0
Fork 0
mirror of https://github.com/deepspeedai/DeepSpeed.git synced 2025-10-20 15:33:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix wandb.log() call by removing sync kwarg (#7383)

Browse Source 
Remove `sync` kwarg from `wandb.log()` invocation, which was [removed in
wandb==0.20.0](https://github.com/wandb/wandb/releases/tag/v0.20.0).

Fixes #7381
This commit is contained in:
Ned Letcher
2025-06-24 02:07:35 +10:00
committed by GitHub
parent d5f6915104
commit 8dd215162b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
deepspeed/monitor/wandb.py
View File

@ -24,10 +24,10 @@ class WandbMonitor(Monitor):
if self.enabled and dist.get_rank() == 0:
wandb.init(project=self.project, group=self.group, entity=self.team)
def log(self, data, step=None, commit=None, sync=None):
def log(self, data, step=None, commit=None):
if self.enabled and dist.get_rank() == 0:
import wandb
return wandb.log(data, step=step, commit=commit, sync=sync)
return wandb.log(data, step=step, commit=commit)
def write_events(self, event_list):
if self.enabled and dist.get_rank() == 0: