fix python encoder

nagyist · Dec 6, 2017 · 4d10d39 · 4d10d39
1 parent 953b4d7
commit 4d10d39
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/python/encoder.py b/python/encoder.py
@@ -1,3 +1,9 @@
+#!/usr/bin/env python
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
 import os
 import h5py
 import numpy as np
@@ -70,7 +76,7 @@ def serialize(self):
                 w.layer_name = layer_name
                 w.weight_name = weight_name
                 w.shape.extend(list(weight_value.shape))
-                if quantize:
+                if self.quantize:
                     w.type = 'uint8'
                     quantized, min_val, max_val = quantize_arr(weight_value)
                     w.data = quantized.astype(np.uint8).tobytes()