p2pfl.learning.frameworks.pytorch.callbacks package¶ Pytorch callbacks for federated learning. Submodules¶ p2pfl.learning.frameworks.pytorch.callbacks.fedprox_callback module FedProxCallback FedProxCallback.get_name() FedProxCallback.on_before_optimizer_step() FedProxCallback.on_train_start() p2pfl.learning.frameworks.pytorch.callbacks.scaffold_callback module SCAFFOLDCallback SCAFFOLDCallback.get_name() SCAFFOLDCallback.on_before_zero_grad() SCAFFOLDCallback.on_train_batch_start() SCAFFOLDCallback.on_train_end() SCAFFOLDCallback.on_train_start()