explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-25 00:34:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix serialization

Browse Source
This commit is contained in:
Matthew Honnibal 2017-05-29 14:35:53 -05:00
parent 1e6df0a2a1
commit 5b29f227ae
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
spacy/util.py
View File

@ -453,11 +453,12 @@ def model_from_bytes(model, bytes_data):
for layer in queue:
if hasattr(layer, '_mem'):
params = weights[i]
blob = layer._mem._get_blob(params.size)
blob[:] = params
layer._mem._offsets = metas[i]
flat_mem = layer._mem._mem.ravel()
flat_params = params.ravel()
flat_mem[:flat_params.size] = flat_params
layer._mem._offsets.update(metas[i])
if hasattr(layer, '_dims'):
layer._dims[i] = dims[i]
layer._dims.update(dims[i])
i += 1
if hasattr(layer, '_layers'):
queue.extend(layer._layers)