|
- protocol=X.SpeakerDiarization.Custom |
|
- task=Segmentation |
|
- task.batch_size=32 |
|
- task.num_workers=10 |
|
- task.duration=5. |
|
- task.warm_up=0. |
|
- task.loss=bce |
|
- task.vad_loss=bce |
|
- patience=20 |
|
- model=PyanNet |
|
- +model.sincnet.stride=10 |
|
- +model.lstm.num_layers=4 |
|
- +model.lstm.monolithic=True |
|
- +model.lstm.dropout=0.5 |
|
- +model.linear.num_layers=2 |
|
- optimizer=Adam |
|
- optimizer.lr=0.001 |
|
- trainer.benchmark=True |
|
- trainer.gradient_clip_val=0.5 |
|
- trainer.gpus=-1 |
|
- trainer.accelerator=ddp |
|
- +augmentation=background |
|
|