INNER CODE UNIT · Python

_group_lstm_params

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

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(
                'Unsupported RNN builder %r: only models saved with DyNet '
                'VanillaLSTMBuilder (as used by nagisa.fit) are supported.'
                % builder)
        builders.setdefault(builder, []).append((param_id, arr))

    lstms = []
    for builder, plist in builders.items():
        if [pid for pid, _ in plist] != ['_0', '_1', '_2']:

View source record →