Scaffold 基于fedavg方法的改进,代码复现(联邦学习)
当前有个工作需要实现scaffold算法,该方法通过添加修正项c来解决客户端漂移现象,在参考github上的相关框架后,复现了该算法。算法分为三个模块:optimizer: 重写优化器sdgclientscaffold:客户端操作serverscaffold:服务端操作optimizer部分代码:import torchfrom torch.optim import Optimizerclass SCAFFOLDOptimizer(Optimizer): def __init__
原创
2021-11-30 18:03:52 ·
2925 阅读 ·
2 评论