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))
评论列表
文章目录