From 236a62af5cd7fb19ae8188e36cb6b9712d938f93 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Thu, 29 Sep 2022 15:52:28 +0800 Subject: [PATCH] Update Signed-off-by: Jael Gu --- nn_fingerprint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nn_fingerprint.py b/nn_fingerprint.py index 27925f3..8c1c5c3 100644 --- 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