def METHOD_NAME(self, save_path):
    """Saves the model to the given path."""
    weights_save_path = os.path.join(save_path, MODEL_WEIGHTS_FILE_NAME)
    torch.METHOD_NAME(self.state_dict(), weights_save_path)