config.json

{
  "depth": 6,
  "dim": 256,
  "embed_dim": 256,
  "heads": 8,
  "max_crossings": 128,
  "target_mean": [
    4.885252952575684,
    17.482948303222656
  ],
  "target_std": [
    0.9845057725906372,
    3.519686460494995
  ]
}