github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-05 18:14:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

disable wandb tb syncing

Browse Source
This commit is contained in:
YeAnbang 2025-05-15 16:52:31 +08:00
parent 4ec73298b2
commit 957e3a521a
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
applications/ColossalChat/coati/distributed/grpo_consumer.py
View File

@ -153,7 +153,7 @@ class GRPOConsumer(BaseConsumer):
):
self.wandb_run = wandb.init(
project=self.project_name,
sync_tensorboard=True,
sync_tensorboard=False,
dir="./wandb",
name=self.run_name,
group=self.wandb_group_name,

2
applications/ColossalChat/coati/distributed/producer.py
View File

@ -71,7 +71,7 @@ class BaseProducer:
if self.producer_idx == 0:
self.wandb_run = wandb.init(
project=project_name,
sync_tensorboard=True,
sync_tensorboard=False,
dir="./wandb",
name=run_name + "_eval",
group=wandb_group_name,