p2pfl.stages.base_node.train_stage module¶
Train stage.
- class p2pfl.stages.base_node.train_stage.TrainStage[source]¶
Bases:
StageTrain stage.
- static execute(state=None, communication_protocol=None, learner=None, aggregator=None, **kwargs)[source]¶
Execute the stage.
- Parameters:
state (
NodeState|None)communication_protocol (
CommunicationProtocol|None)learner (
Learner|None)aggregator (
Aggregator|None)
- Return type:
type[Stage] |None