Skip to content

num_generator

desperadoccy edited this page May 28, 2024 · 3 revisions

num_generator模块介绍

num_generator模块主要用于半异步联邦学习,根据客户端的数据,生成一个聚合数,用于确定本轮参与聚合的更新数量。

  • path: 该聚合数生成类的路径
  • params: 初始化该类所需的params 亦可为一个整数

NumGeneratorFactory

工厂类,用于生成NumGenerator类。如配置的"num_generator": 1之类,将会生成StaticNumGenerator类,否则根据path字段生成对应的类。

现有类介绍

本节下介绍的参数填写在params字段中

AbstractNumGenerator

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

StaticNumGenerator

每轮生成一个由用户指定nums的固定数值。

nums: 一个整数,表示每轮聚合更新数。

Clone this wiki locally