config.json

{
  "architectures": [
    "CodelorColor"
  ],
  "model_type": "codelor-color",
  "format": "codelor_color_standalone_v2",
  "version": "1.2",
  "task": "image-segmentation",
  "labels": [
    "background",
    "green"
  ],
  "green_ref": [
    0.2,
    0.85,
    0.3
  ],
  "hyperparameters": {
    "dim": 128,
    "depth": 8,
    "heads": 8,
    "tied_after": 4,
    "mlp_ratio": 3.0,
    "max_tokens": 1024
  },
  "training": {
    "best_loss": 0.17389452453454335,
    "epochs_done": 30,
    "params": 1641055
  }
}