- 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 | |