Skip to content

updater

desperadoccy edited this page May 28, 2024 · 2 revisions

updater模块介绍

updater模块主要负责进行模型的聚合更新过程。 wandb是否开启可通过self.config['enabled']进行判断。

  • path: updater模块的路径
  • update: 聚合算法, 用于指定聚合算法
  • loss: 损失函数, 用于指定损失函数

现有类介绍

BaseUpdater

该类为基类,定义了必须实现的函数,以及基本属性

AsyncUpdater

异步聚合器,用于异步聚合更新

  • num_generator: int|dict, 每次聚合的更新数量

SyncUpdater

同步聚合器,用于同步聚合更新

SemiASyncUpdater

半异步聚合器,用于半异步聚合更新

  • group: dict, 用于组间的聚合算法

PersonalUpdater

基于同步的个性化聚合器,用于个性化聚合更新,会在每次聚合后,计算本轮的平均精度

ActiveASyncUpdater

基于异步的主动聚合器,会在每次聚合后,主动上传最新的全局模型,以便客户端获取。

Clone this wiki locally