File size: 1,391 Bytes
47695c8 | 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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | {
"format": "lightwam_ad_v1",
"epoch": 100,
"step": 131700,
"checkpoint": "step_131700.pt",
"sha256": "7b880759c6c4244cb9123cfd9320eed303dd9b6204979733fbfe4c96b95c52c5",
"base_model": "Wan-AI/Wan2.1-T2V-1.3B",
"task": "lightwam_ad_navsim_front_384x672",
"tap_layers": [
8,
16,
24
],
"backbone_lora": {
"enabled": true,
"layer_indices": [
0,
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
],
"target_modules": [
"self_attn.q",
"self_attn.k",
"self_attn.v",
"self_attn.o",
"cross_attn.q",
"cross_attn.k",
"cross_attn.v",
"cross_attn.o",
"ffn.0",
"ffn.2"
],
"rank": 64,
"alpha": 128.0,
"dropout": 0.0
},
"training": {
"epochs": 100,
"effective_batch_size": 64,
"precision": "bf16",
"learning_rate": 0.0001,
"optimizer": "AdamW",
"weight_decay": 0.01,
"lr_scheduler": "cosine",
"seed": 42,
"train_clips": 84258,
"validation_clips": 851
},
"navsim_v2_epdms_100": {
"navtest": 88.576,
"navhard": 33.281
},
"code": "https://github.com/L1ziang/LightWAM-AD/tree/lightwam-ad"
}
|