config.json

{
  "architectures": [
    "EcapaModel"
  ],
  "attention_channels": 128,
  "auto_map": {
    "AutoConfig": "configuration_ecapa.EcapaConfig",
    "AutoModel": "modeling_ecapa.EcapaModel"
  },
  "channels": [
    512,
    512,
    512,
    512,
    1536
  ],
  "dilations": [
    1,
    2,
    3,
    4,
    1
  ],
  "global_context": true,
  "groups": [
    1,
    1,
    1,
    1,
    1
  ],
  "hidden_size": 192,
  "hop_length": 10,
  "initializer_range": 0.02,
  "kernel_sizes": [
    5,
    3,
    3,
    3,
    1
  ],
  "loss_fn": "aam",
  "mean_norm": true,
  "model_type": "ecapa",
  "n_mels": 80,
  "norm_type": "sentence",
  "num_classes": 5994,
  "res2net_scale": 8,
  "sample_rate": 16000,
  "se_channels": 128,
  "std_norm": false,
  "torch_dtype": "float32",
  "transformers_version": "4.31.0",
  "win_length": 25
}