- 支持
CPU训练以及基于CUDA平台的计算加速
- 2024/10/21 更新：1.重写了FedAvg-dataAvg.py从server到client的参数返回控制流，将其从server类中分离出来并在模型聚合后实现，使其更加符合论文原始逻辑；2.完善了绘制图像部分的代码，解决了部分参数未使用.item()方法从张量转化为标量的问题；3.更新了可视化流程与cuda加速相关的检测。
- 持续更新ing
-
Notifications
You must be signed in to change notification settings - Fork 0
联邦学习相关算法及代码实现
License
SowingG2333/federated-learning
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
联邦学习相关算法及代码实现
Topics
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published