fl_server_ai.trainer.events.model_test_finished ¶

Bases: ModelTrainerEvent

              flowchart TD
              fl_server_ai.trainer.events.model_test_finished.ModelTestFinished[ModelTestFinished]
              fl_server_ai.trainer.events.base.ModelTrainerEvent[ModelTrainerEvent]

                              fl_server_ai.trainer.events.base.ModelTrainerEvent --> fl_server_ai.trainer.events.model_test_finished.ModelTestFinished
                


              click fl_server_ai.trainer.events.model_test_finished.ModelTestFinished href "" "fl_server_ai.trainer.events.model_test_finished.ModelTestFinished"
              click fl_server_ai.trainer.events.base.ModelTrainerEvent href "" "fl_server_ai.trainer.events.base.ModelTrainerEvent"
            

Model test finished event.

Methods:

 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21

class ModelTestFinished(ModelTrainerEvent):
    """
    Model test finished event.
    """

    def next(self):
        if self.training.model.round < self.training.target_num_updates:
            self.trainer.start_round()
        else:
            self.trainer.finish()

    def handle(self):
        # currently do nothing
        # Potentially, one could aggregate all common metrics here.
        pass

handle()

18
19
20
21

def handle(self):
    # currently do nothing
    # Potentially, one could aggregate all common metrics here.
    pass

next()

12
13
14
15
16

def next(self):
    if self.training.model.round < self.training.target_num_updates:
        self.trainer.start_round()
    else:
        self.trainer.finish()