雾聪
2023-08-07 f8d1c79fe355efb18ae49e4363307dfec3ab89ce
egs/callhome/eend_ola/conf/train_diar_eend_ola_simu_allspkr_chunk2000.yaml
New file
@@ -0,0 +1,44 @@
# network architecture
# encoder related
encoder: eend_ola_transformer
encoder_conf:
    idim: 345
    n_layers: 4
    n_units: 256
# encoder-decoder attractor related
encoder_decoder_attractor: eda
encoder_decoder_attractor_conf:
    n_units: 256
# model related
model: eend_ola
model_conf:
    max_n_speaker: 8
# optimization related
accum_grad: 1
grad_clip: 5
max_epoch: 1
val_scheduler_criterion:
    - valid
    - loss
best_model_criterion:
-   - valid
    - loss
    - min
keep_nbest_models: 100
optim: adam
optim_conf:
    lr: 0.00001
dataset_conf:
    data_names: speech_speaker_labels
    data_types: kaldi_ark
    batch_conf:
        batch_type: unsorted
        batch_size: 8
    num_workers: 8
log_interval: 50