p2pfl.learning.frameworks.pytorch package¶ Pytorch integration for p2pfl. Subpackages¶ p2pfl.learning.frameworks.pytorch.callbacks package Submodules p2pfl.learning.frameworks.pytorch.callbacks.scaffold_callback module SCAFFOLDCallback Submodules¶ p2pfl.learning.frameworks.pytorch.lightning_dataset module PyTorchExportStrategy PyTorchExportStrategy.export() TorchvisionDatasetFactory TorchvisionDatasetFactory.get_mnist() p2pfl.learning.frameworks.pytorch.lightning_learner module LightningLearner LightningLearner.evaluate() LightningLearner.fit() LightningLearner.get_framework() LightningLearner.interrupt_fit() LightningLearner.set_addr() p2pfl.learning.frameworks.pytorch.lightning_logger module FederatedLogger FederatedLogger.finalize() FederatedLogger.log_hyperparams() FederatedLogger.log_metrics() FederatedLogger.name FederatedLogger.save() FederatedLogger.version p2pfl.learning.frameworks.pytorch.lightning_model module LightningModel LightningModel.get_framework() LightningModel.get_parameters() LightningModel.set_parameters()