今天在跑melgan时候,发现GPU使用率低的感人。。
原本以为是pytorch的dataloader的问题
之后把num_worker改为0,用主进程进行读取数据。
发现主要卡在了librosa的load上
from librosa.core import load
data, sr =load('test.wav', sr=16000)
做了个小测试如下:
原来在librosa 0.6版本的时候,不依赖于soundfile。读取很慢,0.7及以上,就使用soundfile了。。
把以上代码改为
import soundfile as sf
data, sr = sf.read('test.wav')
速度可以直接提上去,或者安装高版本的librosa