diff --git a/trainer/train_agent.py b/trainer/train_agent.py
index 468fbf9..772022e 100644
--- a/trainer/train_agent.py
+++ b/trainer/train_agent.py
@@ -427,7 +427,7 @@ if __name__ == "__main__":
 
     wandb = None
     if args.use_wandb and is_main_process():
-        import swanlab as wandb
+        import wandb as wandb
         wandb_id = ckp_data.get('wandb_id') if ckp_data else None
         resume = 'must' if wandb_id else None
         wandb.init(project=args.wandb_project, name=f"Agent-RL-E{args.epochs}-B{args.batch_size}-LR{args.learning_rate}", id=wandb_id, resume=resume)
diff --git a/trainer/train_distillation.py b/trainer/train_distillation.py
index 8e8714c..5182d24 100644
--- a/trainer/train_distillation.py
+++ b/trainer/train_distillation.py
@@ -195,7 +195,7 @@ if __name__ == "__main__":
     # ========== 4. 配wandb ==========
     wandb = None
     if args.use_wandb and is_main_process():
-        import swanlab as wandb
+        import wandb as wandb
         wandb_id = ckp_data.get('wandb_id') if ckp_data else None
         resume = 'must' if wandb_id else None
         wandb_run_name = f"MiniMind-Distill-S{args.student_hidden_size}T{args.teacher_hidden_size}-Epoch-{args.epochs}-BS-{args.batch_size}-LR-{args.learning_rate}"
diff --git a/trainer/train_dpo.py b/trainer/train_dpo.py
index d541c43..987ccc8 100644
--- a/trainer/train_dpo.py
+++ b/trainer/train_dpo.py
@@ -173,7 +173,7 @@ if __name__ == "__main__":
     # ========== 4. 配wandb ==========
     wandb = None
     if args.use_wandb and is_main_process():
-        import swanlab as wandb
+        import wandb as wandb
         wandb_id = ckp_data.get('wandb_id') if ckp_data else None
         resume = 'must' if wandb_id else None
         wandb_run_name = f"MiniMind-DPO-Epoch-{args.epochs}-BatchSize-{args.batch_size}-LR-{args.learning_rate}"
diff --git a/trainer/train_full_sft.py b/trainer/train_full_sft.py
index 4cef014..7a2867e 100644
--- a/trainer/train_full_sft.py
+++ b/trainer/train_full_sft.py
@@ -125,7 +125,7 @@ if __name__ == "__main__":
     # ========== 4. 配wandb ==========
     wandb = None
     if args.use_wandb and is_main_process():
-        import swanlab as wandb
+        import wandb as wandb
         wandb_id = ckp_data.get('wandb_id') if ckp_data else None
         resume = 'must' if wandb_id else None
         wandb_run_name = f"MiniMind-Full-SFT-Epoch-{args.epochs}-BatchSize-{args.batch_size}-LearningRate-{args.learning_rate}"
diff --git a/trainer/train_grpo.py b/trainer/train_grpo.py
index d39f0d6..c1f4d8d 100755
--- a/trainer/train_grpo.py
+++ b/trainer/train_grpo.py
@@ -262,7 +262,7 @@ if __name__ == "__main__":
     # ========== 4. 配wandb ==========
     wandb = None
     if args.use_wandb and is_main_process():
-        import swanlab as wandb
+        import wandb as wandb
         wandb_id = ckp_data.get('wandb_id') if ckp_data else None
         resume = 'must' if wandb_id else None
         wandb_run_name = f"MiniMind-GRPO-Epoch-{args.epochs}-BS-{args.batch_size}-LR-{args.learning_rate}"
diff --git a/trainer/train_lora.py b/trainer/train_lora.py
index 9f4c64b..94c1094 100644
--- a/trainer/train_lora.py
+++ b/trainer/train_lora.py
@@ -119,7 +119,7 @@ if __name__ == "__main__":
     # ========== 4. 配wandb ==========
     wandb = None
     if args.use_wandb and is_main_process():
-        import swanlab as wandb
+        import wandb as wandb
         wandb_id = ckp_data.get('wandb_id') if ckp_data else None
         resume = 'must' if wandb_id else None
         wandb_run_name = f"MiniMind-LoRA-{args.lora_name}-Epoch-{args.epochs}-BatchSize-{args.batch_size}-LR-{args.learning_rate}"
diff --git a/trainer/train_ppo.py b/trainer/train_ppo.py
index 07ba5ac..3fcda7d 100644
--- a/trainer/train_ppo.py
+++ b/trainer/train_ppo.py
@@ -356,7 +356,7 @@ if __name__ == "__main__":
     # ========== 4. 配wandb ==========
     wandb = None
     if args.use_wandb and is_main_process():
-        import swanlab as wandb
+        import wandb as wandb
         wandb_id = ckp_data.get('wandb_id') if ckp_data else None
         resume = 'must' if wandb_id else None
         wandb_run_name = f"MiniMind-PPO-Epoch-{args.epochs}-BS-{args.batch_size}-LR-{args.learning_rate}"
diff --git a/trainer/train_pretrain.py b/trainer/train_pretrain.py
index 13f67b0..73e9045 100644
--- a/trainer/train_pretrain.py
+++ b/trainer/train_pretrain.py
@@ -124,7 +124,7 @@ if __name__ == "__main__":
     # ========== 4. 配wandb ==========
     wandb = None
     if args.use_wandb and is_main_process():
-        import swanlab as wandb
+        import wandb as wandb
         wandb_id = ckp_data.get('wandb_id') if ckp_data else None
         resume = 'must' if wandb_id else None
         wandb_run_name = f"MiniMind-Pretrain-Epoch-{args.epochs}-BatchSize-{args.batch_size}-LearningRate-{args.learning_rate}"