p2pfl.learning.frameworks.pytorch.lightning_learner module

Lightning Learner for P2PFL.

class p2pfl.learning.frameworks.pytorch.lightning_learner.LightningLearner(model=None, data=None, aggregator=None)[source]

Bases: Learner

Learner with PyTorch Lightning.

Parameters:

  • model (Optional[P2PFLModel]) – The model of the learner.

  • data (Optional[P2PFLDataset]) – The data of the learner.

  • addr – The address of the learner.

evaluate(*args: Any, **kwargs: Any) → Any
Return type:

Any

fit(*args: Any, **kwargs: Any) → Any
Return type:

Any

get_framework(*args: Any, **kwargs: Any) → Any
Return type:

Any

interrupt_fit(*args: Any, **kwargs: Any) → Any
Return type:

Any

set_addr(addr)[source]

Set the addr of the node.

Return type:

str