From bd7cb60a57a75470011fec56c6b2dc241d6f69fb Mon Sep 17 00:00:00 2001 From: jiangjiajun Date: Mon, 22 Jul 2019 18:16:57 +0800 Subject: [PATCH] try to fix bug in win --- tensorflow2fluid/tf2fluid/paddle_emitter.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/tensorflow2fluid/tf2fluid/paddle_emitter.py b/tensorflow2fluid/tf2fluid/paddle_emitter.py index 42a98d2..7a6d355 100644 --- a/tensorflow2fluid/tf2fluid/paddle_emitter.py +++ b/tensorflow2fluid/tf2fluid/paddle_emitter.py @@ -22,6 +22,7 @@ import logging import math import struct import numpy +import os logging.basicConfig(level=logging.DEBUG) @@ -166,10 +167,10 @@ class PaddleEmitter(object): "float64": "d" } shape = weight.shape - filew = open(dir + "/" + paddle_var_name, "wb") - filew.write(struct.pack('i', 0)) - filew.write(struct.pack('L', 0)) - filew.write(struct.pack('i', 0)) + filew = open(os.path.join(dir, paddle_var_name), "wb") + numpy.array([0], dtype=numpy.int32).tofile(filew) + numpy.array([0], dtype=numpy.int64).tofile(filew) + numpy.array([0], dtype=numpy.int32).tofile(filew) tensor_desc = framework.VarType.TensorDesc() if str(weight.dtype) in numpy_dtype_map: tensor_desc.data_type = numpy_dtype_map[str(weight.dtype)] @@ -177,7 +178,7 @@ class PaddleEmitter(object): raise Exception("Unexpected array dtype [{}]".format(weight.dtype)) tensor_desc.dims.extend(shape) desc_size = tensor_desc.ByteSize() - filew.write(struct.pack('i', desc_size)) + numpy.array([desc_size], dtype=numpy.int32).tofile(filew) filew.write(tensor_desc.SerializeToString()) weight.tofile(filew) filew.close() -- GitLab