Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · 236a62af5c
1 changed files with 1 additions and 1 deletions
--- a/nn_fingerprint.py
+++ b/nn_fingerprint.py
@ -120,7 +120,7 @@ class NNFingerprint(NNOperator):
                audio = numpy.hstack(frames)
                if len(audio.shape) == 1:
                    audio = audio[None, :]
            audio = self.int2float(audio, dtype='float32')
            audio = self.int2float(audio)
            audio = torch.from_numpy(audio)
        assert len(audio.shape) == 2