p2pfl.learning.frameworks package

Package for the learning frameworks in P2PFL.

This package contains all the integrations with the different learning frameworks.

class p2pfl.learning.frameworks.Framework(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: Enum

Enum for the different learning frameworks.

FLAX = 'flax'

PYTORCH = 'pytorch'

TENSORFLOW = 'tensorflow:='

Subpackages

• p2pfl.learning.frameworks.flax package
  • Submodules
    • p2pfl.learning.frameworks.flax.flax_dataset module
      • FlaxExportStrategy
    • p2pfl.learning.frameworks.flax.flax_learner module
      • FlaxLearner
    • p2pfl.learning.frameworks.flax.flax_model module
      • FlaxModel
      • MLP
• p2pfl.learning.frameworks.pytorch package
  • Subpackages
    • p2pfl.learning.frameworks.pytorch.callbacks package
      • Submodules
  • Submodules
    • p2pfl.learning.frameworks.pytorch.lightning_dataset module
      • PyTorchExportStrategy
      • TorchvisionDatasetFactory
    • p2pfl.learning.frameworks.pytorch.lightning_learner module
      • LightningLearner
    • p2pfl.learning.frameworks.pytorch.lightning_logger module
      • FederatedLogger
    • p2pfl.learning.frameworks.pytorch.lightning_model module
      • LightningModel
      • MLP
• p2pfl.learning.frameworks.simulation package
  • try_init_learner_with_ray()
  • Submodules
    • p2pfl.learning.frameworks.simulation.actor_pool module
      • SuperActorPool
    • p2pfl.learning.frameworks.simulation.utils module
      • check_client_resources()
      • pool_size_from_resources()
    • p2pfl.learning.frameworks.simulation.virtual_learner module
      • VirtualNodeLearner
• p2pfl.learning.frameworks.tensorflow package
  • Subpackages
    • p2pfl.learning.frameworks.tensorflow.callbacks package
      • Submodules
  • Submodules
    • p2pfl.learning.frameworks.tensorflow.keras_dataset module
      • KerasExportStrategy
    • p2pfl.learning.frameworks.tensorflow.keras_learner module
      • KerasLearner
    • p2pfl.learning.frameworks.tensorflow.keras_model module
      • KerasModel
      • MLP
      • ModelNotBuiltError

Submodules

• p2pfl.learning.frameworks.callback module
  • P2PFLCallback
    • P2PFLCallback.get_info()
    • P2PFLCallback.get_name()
    • P2PFLCallback.set_info()
• p2pfl.learning.frameworks.callback_factory module
  • CallbackFactory
    • CallbackFactory.create_callbacks()
    • CallbackFactory.register_callback()
• p2pfl.learning.frameworks.exceptions module
  • DecodingParamsError
  • ModelNotMatchingError
• p2pfl.learning.frameworks.learner module
  • Learner
    • Learner.add_callback_info_to_model()
    • Learner.evaluate()
    • Learner.fit()
    • Learner.get_data()
    • Learner.get_framework()
    • Learner.get_model()
    • Learner.interrupt_fit()
    • Learner.set_addr()
    • Learner.set_data()
    • Learner.set_epochs()
    • Learner.set_model()
    • Learner.update_callbacks_with_model_info()
• p2pfl.learning.frameworks.learner_factory module
  • LearnerFactory
    • LearnerFactory.create_learner()
• p2pfl.learning.frameworks.p2pfl_model module
  • P2PFLModel
    • P2PFLModel.add_info()
    • P2PFLModel.build_copy()
    • P2PFLModel.decode_parameters()
    • P2PFLModel.encode_parameters()
    • P2PFLModel.get_contributors()
    • P2PFLModel.get_framework()
    • P2PFLModel.get_info()
    • P2PFLModel.get_model()
    • P2PFLModel.get_num_samples()
    • P2PFLModel.get_parameters()
    • P2PFLModel.set_contribution()
    • P2PFLModel.set_parameters()