File size: 1,449 Bytes
83360fe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
from transformers.configuration_utils import PretrainedConfig
class SvectorConfig(PretrainedConfig):
model_type = 'svector'
def __init__(
self,
n_mels=40,
sample_rate=16000,
win_length=25,
hop_length=10,
mean_norm=True,
std_norm=False,
norm_type='sentence',
num_heads=8,
num_layers=5,
hidden_size=512,
num_classes=1251,
loss_fn='aam',
auto_map={
"AutoConfig": "configuration_svector.SvectorConfig",
"AutoModel": "modeling_svector.SvectorModel",
"AutoModelForAudioClassification": "modeling_svector.SvectorModelForSequenceClassification"
},
initializer_range=0.02,
**kwargs
):
# Compute features
self.n_mels = n_mels
self.sample_rate = sample_rate
self.win_length = win_length
self.hop_length = hop_length
# Mean variance norm
self.mean_norm = mean_norm
self.std_norm = std_norm
self.norm_type = norm_type
# Embedding model
self.hidden_size = hidden_size
self.num_heads = num_heads
self.num_layers = num_layers
# Classifier
self.num_classes = num_classes
self.loss_fn = loss_fn
# Others
self.auto_map = auto_map
self.initializer_range = initializer_range
super().__init__(**kwargs)
|