search

sp-uhh / sgmse

Score-based Generative Models (Diffusion Models) for Speech Enhancement and Dereverberation
MIT License
454 stars 69 forks source link

What is the recommended hardware for evaluation? #14

Closed djm1176 closed 7 months ago

djm1176 commented 1 year ago

I am running out of GPU memory (8 GB). How much GPU memory is needed to run the model for evaluation?

julius-richter commented 1 year ago

When I run the evaluation script on my GPU it uses 6378 MB of memory. Maybe there are other processes running on your GPU at the same time, because it should actually be possible to run the script with 8 GB of GPU memory.

szh404 commented 8 months ago

I am running out of GPU memory (8 GB). How much GPU memory is needed to run the model for evaluation?

can you run evaluation successfully? If you run the command that [python enhancement.py ...]