File size: 206 Bytes
cb95574
 
 
 
 
 
 
1
2
3
4
5
6
7
8
from transformers import PretrainedConfig

class ECAPAConfig(PretrainedConfig):
    model_type = "ecapa_tdnn"
    def __init__(self, C=1024, **kwargs):
        super().__init__(**kwargs)
        self.C = C