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
553 stars 157 forks source link

# of Epochs for training a FullBand baseline model #22

Closed kadir-gunel closed 3 years ago

kadir-gunel commented 3 years ago

Hello,

My question is about the training. I am just trying to replicate the results with the DNS challange dataset. The number of epochs for the fullband_baseline.toml file is set to 9999 which seems to be a "little" high :) Could you please shed some light on it ? Is this the default value ?

Thank you for sharing your work.

B.R.

haoxiangsnr commented 3 years ago

Hi, Kadir,

Sorry, I’m preparing to graduate recently and I haven’t had time to reply to you.

"9999" is the upper limit number of the training epoch. I think you should use TensorBoard to check whether the model converges during the training.

Thank you for your attention.

kadir-gunel commented 3 years ago

Ok. Thank you.

Good luck with the graduation ! Hope everything goes well :)