import torch


def save_model(network, path):
    """
    Save (trained) neural network to a file.

    Intended to save trained models, but can save untrained ones as well.

    Parameters
    ----------
    network: pytorch object
        Pytorch object or custom pytorch object containing model information.
    path: str
        Location for saving the file, including file name.

    Returns
    -------
    None
    """
    PATH = path
    torch.save(network.state_dict(), PATH)
    return None