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