Update serialization for mem change

explosion · Oct 13, 2017 · 2a1ec8b · 2a1ec8b
1 parent faf86b1
commit 2a1ec8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thinc/neural/_classes/model.py b/thinc/neural/_classes/model.py
@@ -309,7 +309,7 @@ def to_bytes(self):
                     weights[-1][b'seed'] = layer.seed
 
                 offsets = sorted(layer._mem._offsets.items())
-                for (id_, name), (start, row, shape) in offsets:
+                for (id_, name), (start, row, shape, size) in offsets:
                     if row == 1:
                         continue
                     param = layer._mem.get((id_, name))