You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.2 KiB
36 lines
1.2 KiB
3 years ago
|
###########################################################
|
||
|
|
# FEATURE EXTRACTION SETTING #
|
||
|
|
###########################################################
|
||
|
|
# currently, we only support fbank
|
||
|
|
feature:
|
||
|
|
n_mels: 80
|
||
|
|
window_size: 400 #25ms, sample rate 16000, 25 * 16000 / 1000 = 400
|
||
|
|
hop_length: 160 #10ms, sample rate 16000, 10 * 16000 / 1000 = 160
|
||
|
|
|
||
|
|
|
||
|
|
###########################################################
|
||
|
|
# MODEL SETTING #
|
||
|
|
###########################################################
|
||
|
|
# currently, we only support ecapa-tdnn in the ecapa_tdnn.yaml
|
||
|
|
# if we want use another model, please choose another configuration yaml file
|
||
|
|
model:
|
||
|
|
input_size: 80
|
||
|
|
##"channels": [1024, 1024, 1024, 1024, 3072],
|
||
|
|
# "channels": [512, 512, 512, 512, 1536],
|
||
|
|
channels: [512, 512, 512, 512, 1536]
|
||
|
|
kernel_sizes: [5, 3, 3, 3, 1]
|
||
|
|
dilations: [1, 2, 3, 4, 1]
|
||
|
|
attention_channels: 128
|
||
|
|
lin_neurons: 192
|
||
|
|
|
||
|
|
###########################################
|
||
|
|
# Training #
|
||
|
|
###########################################
|
||
|
|
seed: 0
|
||
|
|
epochs: 10
|
||
|
|
batch_size: 32
|
||
|
|
num_workers: 2
|
||
|
|
save_freq: 10
|
||
|
|
log_freq: 10
|
||
|
|
learning_rate: 1e-8
|