3.1.8. aijack.collaborative.fedprox package#

3.1.8.1. Submodules#

3.1.8.2. aijack.collaborative.fedprox.api module#

class aijack.collaborative.fedprox.api.FedProxAPI(*args, mu=0.01, **kwargs)[source]#

Bases: aijack.collaborative.fedavg.api.FedAVGAPI

Implementation of FedProx (https://arxiv.org/abs/1812.06127)

local_train(i)[source]#
run()[source]#
class aijack.collaborative.fedprox.api.MPIFedProxAPI(*args, mu=0.01, **kwargs)[source]#

Bases: aijack.collaborative.fedavg.api.MPIFedAVGAPI

local_train(com_cnt)[source]#

3.1.8.3. aijack.collaborative.fedprox.client module#

class aijack.collaborative.fedprox.client.FedProxClient(model, user_id=0, lr=0.1, send_gradient=True, optimizer_type_for_global_grad='sgd', server_side_update=True, optimizer_kwargs_for_global_grad={}, device='cpu')[source]#

Bases: aijack.collaborative.fedavg.client.FedAVGClient

local_train(server_parameters, local_epoch, criterion, trainloader, optimizer, communication_id=0)[source]#

3.1.8.4. Module contents#

class aijack.collaborative.fedprox.FedProxAPI(*args, mu=0.01, **kwargs)[source]#

Bases: aijack.collaborative.fedavg.api.FedAVGAPI

Implementation of FedProx (https://arxiv.org/abs/1812.06127)

local_train(i)[source]#
run()[source]#
class aijack.collaborative.fedprox.MPIFedProxAPI(*args, mu=0.01, **kwargs)[source]#

Bases: aijack.collaborative.fedavg.api.MPIFedAVGAPI

local_train(com_cnt)[source]#