File size: 510 Bytes
07d5bbe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
- 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
|