File size: 444 Bytes
32bac05 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import scipy.io.wavfile
def save_sample(file_path, sampling_rate, audio):
"""Helper function to save sample
Args:
file_path (str or pathlib.Path): save file path
sampling_rate (int): sampling rate of audio (usually 22050)
audio (torch.FloatTensor): torch array containing audio in [-1, 1]
"""
audio = (audio.numpy() * 32768).astype("int16")
scipy.io.wavfile.write(file_path, sampling_rate, audio)
|