segmentation / overrides.yaml
salmanshahid's picture
Upload folder using huggingface_hub
07d5bbe verified
raw
history blame contribute delete
510 Bytes
- 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