From 5b29f227aea7c99d89b7fda1ea436e1ed2a6de8f Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Mon, 29 May 2017 14:35:53 -0500 Subject: [PATCH] Fix serialization --- spacy/util.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/spacy/util.py b/spacy/util.py index d75b4f610..955dc4c5d 100644 --- a/spacy/util.py +++ b/spacy/util.py @@ -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)