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()
• 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()
  • MLP
    • MLP.configure_optimizers()
    • MLP.forward()
    • MLP.test_step()
    • MLP.training_step()
    • MLP.validation_step()