wavlm_ssl_sv / configs /wavlm_mhfa_dlg_lc.yaml
theolepage's picture
initial commit
430712c
raw
history blame
476 Bytes
max_frames: 300
max_epoch: 15
batch_size: 120
margin: 0.2
eval_frames: 400
augment: True
## Training details
trainfunc: aamsoftmax
scale: 30
lr_decay: 0.95
pretrained_model_path: WavLM-Base+.pt
weight_finetuning_reg: 0.01
LLRD_factor: 1.0
LR_Transformer: 2e-5
LR_MHFA: 5e-3
## Loss functions
nClasses: 7500
## Load and save
save_path: exp/wavlm_mhfa_dlg_lc
# save_path: exp/wavlm_mhfa_dlg_lc_iter2
## Model definition
model: Baseline.Spk_Encoder
nOut: 256
port: 6754