yeyupiaoling / PPASR

基于PaddlePaddle实现端到端中文语音识别,从入门到实战,超简单的入门案例,超实用的企业项目。支持当前最流行的DeepSpeech2、Conformer、Squeezeformer模型
Apache License 2.0
807 stars 128 forks source link

博主有没有办法解决多线程的问题? #126

Closed wxy0505 closed 1 year ago

wxy0505 commented 1 year ago

我batch_size设置16,线程设置0可以训练,但我线程设置4或者2时,batch_size设置8或者4时都会卡住不动。

wxy0505 commented 1 year ago

而且在卡住的时候我看GPU的利用率是0

yeyupiaoling commented 1 year ago

应该随机的吧,我一般减小batch_size就可以了。或你设置为2 https://github.com/yeyupiaoling/PPASR/blob/855b72ef02c5b03ae365e117687785e4950586de/configs/conformer_online_zh.yml#L42