Function load_and_assign_npz

tensorlayer/files/utils.py:2023–2047

Load model from npz and assign to a network. Parameters ------------- name : str The name of the `.npz` file. network : :class:`Model` The network to be assigned. Examples -------- - See ``tl.files.save_npz``

(name=None, network=None)

def load_and_assign_npz(name=None, network=None):
    """Load model from npz and assign to a network.

    Parameters
    -------------
    name : str
        The name of the `.npz` file.
    network : :class:`Model`
        The network to be assigned.

    Examples
    --------
    - See ``tl.files.save_npz``

    """
    if network is None:
        raise ValueError("network is None.")

    if not os.path.exists(name):
        logging.error("file {} doesn't exist.".format(name))
        return False
    else:
        weights = load_npz(name=name)
        assign_weights(weights, network)
        logging.info("[*] Load {} SUCCESS!".format(name))


def save_npz_dict(save_list=None, name='model.npz'):