Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
utils.py 389 Bytes
Newer Older
hwangjeff's avatar
Add global stats script and new json for LibriSpeech RNN-T training recipe (#2183)  
hwangjeff committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 
import math

import torch
import torchaudio


DECIBEL = 2 * 20 * math.log10(torch.iinfo(torch.int16).max)
GAIN = pow(10, 0.05 * DECIBEL)
spectrogram_transform = torchaudio.transforms.MelSpectrogram(sample_rate=16000, n_fft=400, n_mels=80, hop_length=160)


def piecewise_linear_log(x):
    x[x > math.e] = torch.log(x[x > math.e])
    x[x <= math.e] = x[x <= math.e] / math.e
    return x