ModuleNotFoundError: No module named 'torchmetrics.audio.utils'

Audio-WestlakeU / NBSS

The official repo of NBC & SpatialNet for multichannel speech separation, denoising, and dereverberation

MIT License

175 stars 21 forks source link

Closed ermu-tech closed 2 years ago

ermu-tech commented 2 years ago

When I train NBSS, there's something wrong in "NBSS\NBSS-main\models\NBSS_ifp.py", line 15 : ModuleNotFoundError: No module named 'torchmetrics.audio.utils' It seems this file (torchmetrics.audio.utils) doesn't exist?
In "NBSS\NBSS-main\generate_rirs.py", line 18, I couldn't import gpuRIR on Windows 10, how can I get gpuRIR?

quancs commented 2 years ago

Sorry, I will update the code later.

quancs commented 2 years ago

1 is fixed @ermu-tech

quancs commented 2 years ago

for 2, you could find gpuRIR in its official repo: https://github.com/DavidDiazGuerra/gpuRIR

quancs commented 2 years ago

check the new requirements to install Torchmetrics, PytorchLightning and gpuRIR

ermu-tech commented 2 years ago

Thank you so much! I'll try again!