数据集测试样例配置

2025-04-21 22:27:19 +08:00
parent 132ed64136
commit 8cd6df4527
1 changed files with 2 additions and 2 deletions
--- 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数据路径
    ]
    
    # 运行联邦训练