def save_checkpoint()

in src/helpers.py [0:0]

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

    def save_checkpoint(self, model: nn.Module) -> None:
        """Saves model when the metric on the validation set gets improved."""
        torch.save({"model_state_dict": model.state_dict()}, self.filename)