{
 "architecture": "HTDemucs (demucs 4.1.0, MIT)",
 "sources": [
  "drums",
  "bass",
  "other",
  "vocals"
 ],
 "samplerate": 44100,
 "audio_channels": 2,
 "model_cfg": {
  "channels": 44,
  "t_layers": 5,
  "t_heads": 8,
  "bottom_channels": 0,
  "segment": 6.0
 },
 "model_name": "medium",
 "nparams": 22612960,
 "checkpoint": "best",
 "step": 120450,
 "valid_sdr": {
  "drums": 11.869,
  "bass": 8.689,
  "other": 2.352,
  "vocals": 10.811,
  "mean": 8.43,
  "n_songs": 44
 },
 "valid_sdr_mix": {
  "drums": 7.507,
  "bass": 8.648,
  "other": 0.749,
  "vocals": 5.304,
  "mean": 5.552,
  "n_songs": 41
 },
 "provenance": "own weights: random init, trained on the user's Suno stems (sum-of-stems mixtures); stage 2: fine-tuned from our stage-1 weights (best.pt, step 85702, sha256 ad21ed8115643425\u2026) with Suno's delivered mixes as part of the inputs and a multi-resolution STFT loss",
 "load": "separation_train.build_model(model_cfg, sources).load_state_dict(torch.load('weights.pth'))",
 "input_normalisation": "x = (mix - mean(mono)) / std(mono); output * std + mean (as demucs apply)",
 "onnx": {
  "file": "model_fp16.onnx",
  "input": "mix",
  "shape": [
   1,
   4,
   2,
   264600
  ],
  "max_abs_diff": 0.0002949833869934082
 }
}