config.json

{
  "embed_dim": 768,
  "num_classes": 150,
  "dropout": 0.1,
  "use_ln": false,
  "metadata": {
    "resolution": 192,
    "model": "facebook/dinov3-vitb16-pretrain-lvd1689m",
    "best_miou": 0.35755154490470886,
    "step": 30000,
    "config": {
      "ade20k_root": "/local/scratch/yberreby.8741771.0/ADEChallengeData2016",
      "image_size": 512,
      "batch_size": 16,
      "eval_batch_size": 32,
      "num_workers": 4,
      "peak_lr": 0.0003,
      "weight_decay": 0.001,
      "warmup_steps": 1500,
      "warmup_lr_ratio": 1e-06,
      "max_steps": 40000,
      "grad_clip": "inf",
      "dropout": 0.1,
      "aug_scale_range": [
        0.5,
        2.0
      ],
      "aug_flip_prob": 0.5,
      "log_every": 20,
      "val_every": 500,
      "viz_every": 500,
      "viz_samples": 4,
      "comet_project": "canvit-ade20k-probes",
      "comet_workspace": "m2b3-ava",
      "device": "cuda",
      "amp": true,
      "probe_ckpt_dir": "/home/yberreby/projects/def-skrishna/checkpoints/canvit-ade20k-probes",
      "resolution": 192,
      "model": "facebook/dinov3-vitb16-pretrain-lvd1689m"
    }
  }
}