ecapa-aam / config.json
yangwang825's picture
Training in progress (epoch 2)
25aea84
{
"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
}