def load_checkpoint()

in src/helpers.py [0:0]

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

    def load_checkpoint(self, model: nn.Module) -> nn.Module:
        """Load model saved with early stopping."""
        model.load_state_dict(torch.load(self.filename)["model_state_dict"])