Anchor-x/Graduation-Project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

数据集测试样例配置

Browse Source
This commit is contained in:
myh 2025-04-21 22:27:19 +08:00
parent 132ed64136
commit 8cd6df4527
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
federated_learning/yolov8_fed.py
View File

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