p2pfl.stages.base_node.train_stage moduleΒΆ

Train stage.

class p2pfl.stages.base_node.train_stage.TrainStage[source]ΒΆ

Bases: Stage

Train stage.

static execute(state=None, communication_protocol=None, learner=None, aggregator=None, **kwargs)[source]ΒΆ

Execute the stage.

Return type:

Optional[Type[Stage]]

static name()[source]ΒΆ

Return the name of the stage.