data.py 文件源码

def loader(self, path):
        feat = np.load(path)

        txt = feat['phonemes'].astype('int64')
        txt = torch.from_numpy(txt)

        audio = feat['audio_features']
        audio = torch.from_numpy(audio)

        spkr = os.path.basename(path).split('_')[0]

        return txt, audio, spkr