config.json 1.05 KB
{
  "model": {
    "embed_dim": 192,
    "channels": 512,
    "n_mels": 128,
    "use_band_split": true
  },
  "data": {
    "source_dataset": "fma",
    "manifests_dir": "data/external_ingested/synthetic_as_open_fixed/fma/manifests",
    "query_duration": 5.0
  },
  "run": {
    "train_command": "/usr/local/miniconda3/bin/python train.py --data data/external_ingested/synthetic_as_open_fixed/fma/manifests --output data/models_open_smoke_fixed --device cpu --epochs 1 --batch-size 2",
    "index_command": "/usr/local/miniconda3/bin/python run_demo.py build-index --data data/external_ingested/synthetic_as_open_fixed/fma/manifests --model data/models_open_smoke_fixed/best_model.pt --output data/index_open_smoke_fixed --device cpu",
    "eval_command": "/usr/local/miniconda3/bin/python evaluate.py --data data/external_ingested/synthetic_as_open_fixed/fma/manifests --model data/models_open_smoke_fixed/best_model.pt --index-prefix data/index_open_smoke_fixed/reference --split test --device cpu --fast-eval --output-json reports/open-smoke-fixed/fma/eval.json"
  }
}