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

关于训练时长 #18

Closed phenex986 closed 3 years ago

phenex986 commented 3 years ago

您好; 我实际用了大概11h的数据用于训练,batch_size=48, 两块k80显卡; 目前训练发现一个epoch用时大概50分钟,按照.toml中epochs=9999的配置,要训练接近一年。? 想问下这种训练速度是正常的么

phenex986 commented 3 years ago

看了下最好的预训练模型,epoch是在58,所以还是看valid的表现,提前中断吧

haoxiangsnr commented 3 years ago

Exactly. You need to check the training processing.