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
}
} |