config.json

{
  "model_type": "apex",
  "mert_model_name": "m-a-p/MERT-v1-95M",
  "layer_indices": [
    2,
    5,
    8,
    -1
  ],
  "segment_sec": 30,
  "seed": 42,
  "input_dim": 768,
  "shared_dims": [
    512,
    256
  ],
  "branch_dims": [
    128,
    64
  ],
  "dropout_shared": 0.3,
  "dropout_branch": 0.1,
  "auto_map": {
    "AutoConfig": "configuration_apex.APEXConfig",
    "AutoModel": "modeling_apex.APEXModel"
  },
  "architectures": [
    "APEXModel"
  ],
  "epoch": 50,
  "val_loss": 398.77835954938615
}