update encoder.py

This commit is contained in:
Leon Chen
2016-09-17 23:40:34 -04:00
parent ecef1c3900
commit d6aa20f353
+1 -1
View File
@@ -45,7 +45,7 @@ class Encoder(object):
bytearr = weight_value.astype(np.float32).tobytes()
self.weights += bytearr
meta['offset'] = offset
meta['length'] = len(bytearr) / 4
meta['length'] = len(bytearr) // 4
meta['shape'] = list(weight_value.shape)
meta['type'] = 'float32'
self.metadata.append(meta)