From 957e3a521aa41d0580c6ed32d77b4a9cf8581210 Mon Sep 17 00:00:00 2001 From: YeAnbang Date: Thu, 15 May 2025 16:52:31 +0800 Subject: [PATCH] disable wandb tb syncing --- applications/ColossalChat/coati/distributed/grpo_consumer.py | 2 +- applications/ColossalChat/coati/distributed/producer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/applications/ColossalChat/coati/distributed/grpo_consumer.py b/applications/ColossalChat/coati/distributed/grpo_consumer.py index 5ad73f45c..7ea2dbf18 100644 --- a/applications/ColossalChat/coati/distributed/grpo_consumer.py +++ b/applications/ColossalChat/coati/distributed/grpo_consumer.py @@ -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, diff --git a/applications/ColossalChat/coati/distributed/producer.py b/applications/ColossalChat/coati/distributed/producer.py index f7c17bf56..8b9452160 100644 --- a/applications/ColossalChat/coati/distributed/producer.py +++ b/applications/ColossalChat/coati/distributed/producer.py @@ -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,