Skip to content

Commit 70c57b1

Browse files
Merge pull request #1137 from onnx/tom/RemoveSpecialCharsFromModelZip
Remove special characters from model zip for large model format
2 parents 38f5a48 + 80bc0d8 commit 70c57b1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tf2onnx/graph.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,9 @@ def get_value_attr(self, external_tensor_storage=None):
111111
return a
112112
if np.product(a.t.dims) > external_tensor_storage.external_tensor_size_threshold:
113113
a = copy.copy(a)
114-
tensor_name = self.name + "_" + str(external_tensor_storage.name_counter)
114+
tensor_name = self.name.strip() + "_" + str(external_tensor_storage.name_counter)
115+
for c in '~"#%&*:<>?/\\{|}':
116+
tensor_name = tensor_name.replace(c, '_')
115117
external_tensor_storage.name_counter += 1
116118
external_tensor_storage.name_to_tensor_data[tensor_name] = a.t.raw_data
117119
external_tensor_storage.node_to_modified_value_attr[self] = a

0 commit comments

Comments
 (0)