mirror of
https://github.com/snakers4/silero-vad.git
synced 2026-02-05 01:49:22 +08:00
16 lines
1.6 KiB
YAML
16 lines
1.6 KiB
YAML
use_torchhub: True # jit модель будет загружена через torchhub, если True, или через pip, если False
|
|
|
|
tune_8k: False # дообучает 16к голову, если False, и 8к голову, если True
|
|
train_dataset_path: 'train_dataset_path.feather' # путь до датасета в формате feather для дообучения, подробности в README
|
|
val_dataset_path: 'val_dataset_path.feather' # путь до датасета в формате feather для валидации, подробности в README
|
|
model_save_path: 'model_save_path.jit' # путь сохранения дообученной модели
|
|
|
|
noise_loss: 0.5 # коэффициент, применяемый к лоссу на неречевых окнах
|
|
max_train_length_sec: 8 # во время тюнинга аудио длиннее будут обрезаны до данного значения
|
|
aug_prob: 0.4 # вероятность применения аугментаций к аудио в процессе дообучения
|
|
|
|
learning_rate: 5e-4 # темп дообучения модели
|
|
batch_size: 384 # размер батча при дообучении и валидации
|
|
num_workers: 4 # количество потоков, используемых для даталоадеров
|
|
num_epochs: 20 # количество эпох дообучения, 1 эпоха = полный прогон тренировочных данных
|
|
device: 'cpu' # cpu или cuda, на чем будет производится дообучение |