def on_train_end()

in realbook/callbacks/utilization.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

    def on_train_end(self, logs: Dict[Any, Any]) -> None:
        """
        Runs at the end of training. Stops the measurement thread and flushes any buffered
        measurements.
        """
        self.finished = True
        self.measurement_thread.join()
        self.tensorboard_writer.flush()