今天在跑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