diff --git a/federated_learning/yolov8_fed.py b/federated_learning/yolov8_fed.py
index c374fa4..95a56f8 100644
--- a/federated_learning/yolov8_fed.py
+++ b/federated_learning/yolov8_fed.py
@@ -113,8 +113,8 @@ def federated_train(num_rounds, clients_data):
 if __name__ == "__main__":
     # 联邦训练配置
     clients_config = [
-        "./config/client1_data.yaml",  # 客户端1数据路径
-        "./config/client2_data.yaml"  # 客户端2数据路径
+        "../dataset/train1/train1.yaml",  # 客户端1数据路径
+        "../dataset/train2/train2.yaml"  # 客户端2数据路径
     ]
     
     # 运行联邦训练