Extract and manually set states

[TemryL]

Is there a way to extract and manually set states with Silero. I see we can reset the states but not sure if we can set them. Thanks

[snakers4]

Yes, with ONNX models

silero-vad/src/silero_vad/utils_vad.py

Lines 78 to 85 in be95df9

	x = torch.cat([self._context, x], dim=1)
	if sr in [8000, 16000]:
	ort_inputs = {'input': x.numpy(), 'state': self._state.numpy(), 'sr': np.array(sr, dtype='int64')}
	ort_outs = self.session.run(None, ort_inputs)
	out, state = ort_outs
	self._state = torch.from_numpy(state)
	else:
	raise ValueError()

silero-vad/src/silero_vad/utils_vad.py

Lines 51 to 55 in be95df9

	def reset_states(self, batch_size=1):
	self._state = torch.zeros((2, batch_size, 128)).float()
	self._context = torch.zeros(0)
	self._last_sr = 0
	self._last_batch_size = 0

With the PyTorch model in latest version it is tucked in.

[snakers4]

As for PyTorch, model._model(x, state) returns state, model._state keeps state.

[TemryL]

awesome, thanks!