style.py 文件源码

def test():
    return
    num_test = 100
    x_cpu = sampler0.memories_test[0:num_test]
    x = Variable(torch.FloatTensor(x_cpu))
    y = generator(x).data.cpu().numpy()

    chunk_size = y.shape[2]
    output_stft = np.zeros([num_test, chunk_size], dtype=np.complex128)
    for i in range(num_test):
        output_flat = y[i]
        output_complex = output_flat[0] + 1j * output_flat[1]
        output_stft[i] = output_complex

    output_wav = stft.istft(output_stft / spectrogram.spec_norm)

    wavfile.write("test.wav", 8000, output_wav.reshape(-1))