saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-20 19:12:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix bugs in sparsify script (#7134)

Browse source 
Fix type and check.
This commit is contained in:
Dmitri Smirnov 2021-03-25 14:53:52 -07:00 committed by GitHub
parent cc0e7bee76
commit 2bf54bcaa2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tools/python/sparsify_initializers.py
View file

@ -16,7 +16,7 @@ from onnx import ModelProto, SparseTensorProto, TensorProto, numpy_helper
logger = logging.getLogger(__name__)
real_types = set((np.float32, np.float64, np.double))
real_types = set((int(TensorProto.FLOAT), int(TensorProto.DOUBLE)))
def parse_arguments():
@ -67,7 +67,7 @@ def convert_tensor_to_sparse(tensor, tolerance): # type: (TensorProto) -> Tuple
else:
for index in range(data_len):
el = tensor_data[index]
if el == 0:
if el != 0:
values.append(el)
indicies.append(index)
nnz_count += 1