File size: 717 Bytes
bd9de4f 5e8590b bd9de4f 5e8590b bd9de4f 5e8590b bd9de4f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | from transformers import PretrainedConfig
class MosNetConfig(PretrainedConfig):
"""
Это конфигурация для модели MosNet.
Она хранит параметры, определяющие архитектуру модели.
"""
model_type = "mosnet"
def __init__(
self,
sample_rate: int = 16000,
fft_size: int = 512,
hop_length: int = 256,
win_length: int = 512,
dropout: float = 0.3,
**kwargs,
):
super().__init__(**kwargs)
self.sample_rate = sample_rate
self.fft_size = fft_size
self.hop_length = hop_length
self.win_length = win_length
self.dropout = dropout |