Audio-WestlakeU / FullSubNet

PyTorch implementation of "FullSubNet: A Full-Band and Sub-Band Fusion Model for Real-Time Single-Channel Speech Enhancement."
https://fullsubnet.readthedocs.io/en/latest/
MIT License
554 stars 158 forks source link

How to use the cpu training model? #64

Open Cagemoney opened 1 year ago

Cagemoney commented 1 year ago

I want to run the code on the cpu before using the GPU.

haoxiangsnr commented 1 year ago

Hi, I'm afraid you'll have to change too much. To understand the model, you may just run the corresponding model.py.

Cagemoney commented 1 year ago

Ok, thank you for your reply.