Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

nnAudio-CPU memory growth #119

Open
will-wiki opened this issue Oct 28, 2022 · 2 comments
Open

nnAudio-CPU memory growth #119

will-wiki opened this issue Oct 28, 2022 · 2 comments

Comments

@will-wiki
Copy link

will-wiki commented Oct 28, 2022

Hello, I am using nnAudio to extract the CQT features of the audio in the following way


from nnAudio import features
cqt_layer = features.CQT2010v2(sr=16000).to(torch.device('cuda'))
data, sr = librosa.load(audio_file, sr=None)
 with torch.no_grad():
        data = torch.tensor(data, dtype=torch.float32).to(torch.device('cuda'))
        cqt = cqt_layer(data)[0].cpu().numpy()

However, the problem is that when the amount of data requested is fixed, increasing concurrency causes memory to keep growing
image

@gudgud96
Copy link
Contributor

gudgud96 commented Feb 5, 2023

How do you increase concurrency? Using torch.multiprocessing?
Maybe you can share your code with concurrency included.

@migperfer
Copy link
Contributor

Hi! @will-wiki , what do you mean by "fixed amount of data"?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants