slices in correct dimension (audio dim), so batch size is not reduced

This commit is contained in:
mhThomsen
2023-04-27 07:37:54 +02:00
parent 5872cffd78
commit e0c2015193

View File

@@ -34,7 +34,7 @@ class OnnxWrapper():
if sr != 16000 and (sr % 16000 == 0):
step = sr // 16000
x = x[::step]
x = x[:,::step]
sr = 16000
if sr not in self.sample_rates: