config: wandb project name -> sora_speedrun

LambdaLabsML · Oct 11, 2024 · 1cf1b74 · 1cf1b74
1 parent 1a818c7
commit 1cf1b74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train.py b/scripts/train.py
@@ -79,7 +79,7 @@ def main():
     if coordinator.is_master():
         tb_writer = create_tensorboard_writer(exp_dir)
         if cfg.get("wandb", False):
-            wandb.init(project="Open-Sora", name=exp_name, config=cfg.to_dict(), dir="./outputs/wandb")
+            wandb.init(project="sora_speedrun", name=exp_name, config=cfg.to_dict(), dir="./outputs/wandb")
 
     # == init ColossalAI booster ==
     plugin = create_colossalai_plugin(