File size: 1,330 Bytes
709a6fa cde1dbf 709a6fa f871a5c 709a6fa | 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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | {
"model": {
"dim": 384,
"depth": 24,
"patch_size": 16,
"proj_factor": 2.0,
"qkv_proj_blocksize": 4,
"num_heads": 4,
"conv_kernel": 4,
"mlp_ratio": 4.0,
"drop_path_rate": 0.05,
"tmoe_blocks": 2,
"num_experts": 4,
"film_interval": 6,
"feat_size": 16,
"use_uncertainty": true
},
"training": {
"phase1": {
"epochs": 300,
"lr": 1e-4,
"batch_size": 32,
"weight_decay": 0.05,
"warmup_epochs": 5,
"grad_clip": 1.0,
"backbone_lr_scale": 0.1,
"acl_ramp_epochs": 50
},
"phase2": {
"epochs": 100,
"lr": 1e-5,
"batch_size": 32,
"backbone_lr_scale": 0.01,
"warmup_epochs": 2
}
},
"data": {
"template_size": 128,
"search_size": 256,
"datasets": ["got10k", "lasot", "trackingnet", "coco", "visdrone", "uavdt", "webuav3m"]
},
"inference": {
"search_scale": 4.0,
"confidence_threshold": 0.3,
"template_update_threshold": 0.8
},
"constraints": {
"max_params_M": 50,
"max_latency_ms": 30,
"max_gflops": 20,
"max_size_mb": 500
}
} |