Solves data type mismatch issue on windows

Solves windows 11 error regarding mismatched data type. Expected data type is int64 and actual data type is int32
This commit is contained in:
kafan1986
2022-11-11 23:46:51 +05:30
committed by GitHub
parent 82d199ff22
commit 25f04dda35

View File

@@ -61,7 +61,7 @@ class OnnxWrapper():
self.reset_states(batch_size)
if sr in [8000, 16000]:
ort_inputs = {'input': x.numpy(), 'h': self._h, 'c': self._c, 'sr': np.array(sr)}
ort_inputs = {'input': x.numpy(), 'h': self._h, 'c': self._c, 'sr': np.array(sr, dtype='int64')}
ort_outs = self.session.run(None, ort_inputs)
out, self._h, self._c = ort_outs
else: