From 98d84364404231c687db283f14441b1dd2c15e2a Mon Sep 17 00:00:00 2001 From: cftang Date: Tue, 26 Feb 2019 01:20:54 +0800 Subject: [PATCH] fix tf_text_graph_common tensor_content type bug --- samples/dnn/tf_text_graph_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/samples/dnn/tf_text_graph_common.py b/samples/dnn/tf_text_graph_common.py index 5a8e624..b46b1d4 100644 --- a/samples/dnn/tf_text_graph_common.py +++ b/samples/dnn/tf_text_graph_common.py @@ -324,6 +324,6 @@ def writeTextGraph(modelPath, outputPath, outNodes): for node in graph_def.node: if node.op == 'Const': if 'value' in node.attr and node.attr['value'].tensor.tensor_content: - node.attr['value'].tensor.tensor_content = '' + node.attr['value'].tensor.tensor_content = b'' tf.train.write_graph(graph_def, "", outputPath, as_text=True) -- 2.7.4