diff --git a/scripts/train.py b/scripts/train.py index 92c43b84..5a0ced83 100644 --- 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(