INNER CODE UNIT · Python
load_dynet_text_model
taishi-i/nagisa · nagisa/dynet_loader.py:35
def load_dynet_text_model(path):
"""Parse a DyNet text-format model file.
return:
- OrderedDict mapping parameter name to (kind, array) where kind is
"Parameter" or "LookupParameter". Plain parameters keep DyNet's
dim layout (e.g. (6, 100)); lookup parameters are returned as
(num_entries, dim), the same layout as LookupParameters.as_array().
"""
params = OrderedDict()
with open(path, 'rb') as f:
while True:
header = f.readline()
if not header:
break
match = _HEADER_RE.match(header.decode('latin-1'))
if match is None:
raise ValueError(