def on_train_end()

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

• 3 lines of code
• 2 McCabe index (conditional complexity)

    def on_train_end(self, logs: Dict[Any, Any]) -> None:
        """
        Runs at the end of training. Only stops the measurement thread if a GPU is
        available.
        """
        if self.enable:
            super().on_train_end(logs)