INNER CODE UNIT · Python

_check_shape

taishi-i/nagisa · nagisa/dynet_loader.py:89

def _check_shape(name, arr, expected):
    if arr.shape != tuple(expected):
        raise ValueError(
            'Model parameter %s has shape %s, expected %s. The model file '
            'does not match the given hyperparameters.'
            % (name, arr.shape, tuple(expected)))


def _group_lstm_params(items, prefix):
    """Group /<prefix>/vanilla-lstm-builder*/_k params into per-builder lists."""
    builders = OrderedDict()
    for name, (kind, arr) in items:
        parts = name.split('/')
        if len(parts) != 4 or parts[1] != prefix:
            raise ValueError('Unexpected LSTM parameter name %r' % name)
        builder, param_id = parts[2], parts[3]
        if 'lstm-builder' in builder and not builder.startswith('vanilla-lstm-builder'):
            raise ValueError(

View source record →