Introduction
Guides
Miscellaneous
Bases: _Loss
_Loss
Compute FedProx loss: a loss penalizing the deviation from global model.
mu – weighting parameter
Forward pass in training.
input (nn.Module) – the local pytorch model
target (nn.Module) – the copy of global pytorch model when local clients received it at the beginning of each local round
FedProx loss term