File size: 759 Bytes
2dabf64
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
{
  "data": {
    "dataset": "scene_parse_150 (ADE20K, 150 classes)",
    "image_size": 224,
    "patch_size": 16,
    "num_classes": 151,
    "ignore_index": 0
  },
  "model": {
    "architecture": "ViT-Segmentation (first-principles backbone + linear decoder head)",
    "embed_dim": 384,
    "depth": 8,
    "num_heads": 6,
    "mlp_ratio": 4.0,
    "dropout": 0.1
  },
  "training": {
    "batch_size": 8,
    "epochs": 5,
    "lr": 0.0003,
    "weight_decay": 0.05,
    "warmup_epochs": 1,
    "num_workers": 2,
    "subset_size": null,
    "device": "cuda if available else cpu"
  },
  "evolution_search": {
    "population_size": 6,
    "generations": 4,
    "elite_k": 2,
    "steps_per_candidate": 30,
    "subset_size": 64,
    "batch_size": 4
  }
}