Merge pull request #1 from kafan1986/kafan1986-patch-1

Solves data type mismatch issue on windows
This commit is contained in:
kafan1986
2022-11-11 23:50:20 +05:30
committed by GitHub

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: