config.json

{
  "model_type": "calmos",
  "architectures": [
    "CalMOS"
  ],
  "auto_map": {
    "AutoConfig": "configuration_calmos.CalMOSConfig",
    "AutoModel": "modeling_calmos.CalMOS"
  },
  "backbone_name": "facebook/wav2vec2-xls-r-300m",
  "num_feature_layers": 25,
  "adapter_input_dim": 1024,
  "adapter_dim": 1024,
  "mlp_input_dim": 1024,
  "mlp_hidden_dim": 1024,
  "mlp_num_layers": 1,
  "mlp_output_size": 1,
  "mlp_dropout": 0.1,
  "target_sr": 16000,
  "mos_min": 1.0,
  "mos_max": 5.0,
  "layer_weight_strategy": "layer_adapter",
  "pooling_strategy": "mean",
  "freeze_backbone": true
}