guychuk commited on
Commit
e72b643
·
verified ·
1 Parent(s): 59963d7

Upload v6-lam0p1-s2_summary.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. v6-lam0p1-s2_summary.json +93 -0
v6-lam0p1-s2_summary.json ADDED
@@ -0,0 +1,93 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tag": "v6-lam0p1-s2",
3
+ "lambda_jepa": 0.1,
4
+ "beta_nll": 0.5,
5
+ "n_params": 7967886,
6
+ "best_fde_m": 1905.556884765625,
7
+ "final": {
8
+ "in_distribution": {
9
+ "ade_m": 1098.5457763671875,
10
+ "fde_m": 1905.556884765625,
11
+ "fde_median_m": 1299.7315673828125,
12
+ "nll_xy_z": -10.303141593933105,
13
+ "coverage_95": 0.967128027681661,
14
+ "n": 8092,
15
+ "miss_rate_500m": 0.8339100346020761,
16
+ "miss_rate_1000m": 0.6051655956500247,
17
+ "miss_rate_2000m": 0.3320563519525457,
18
+ "per_bucket": {
19
+ "delta_30_60": {
20
+ "ade_m": 790.9866333007812,
21
+ "fde_m": 1092.157470703125,
22
+ "coverage_95": 0.9917414721723519,
23
+ "n": 2785
24
+ },
25
+ "delta_60_90": {
26
+ "ade_m": 1086.21533203125,
27
+ "fde_m": 1839.5618896484375,
28
+ "coverage_95": 0.9598100109609061,
29
+ "n": 2737
30
+ },
31
+ "delta_90_120": {
32
+ "ade_m": 1423.2672119140625,
33
+ "fde_m": 2799.07958984375,
34
+ "coverage_95": 0.947215143793229,
35
+ "n": 2747
36
+ }
37
+ }
38
+ },
39
+ "extrap_delta_180": {
40
+ "ade_m": 2843.651611328125,
41
+ "fde_m": 7678.0986328125,
42
+ "fde_median_m": 7317.61669921875,
43
+ "nll_xy_z": -5.38964319229126,
44
+ "coverage_95": 0.45773603559070686,
45
+ "n": 8092,
46
+ "miss_rate_500m": 0.9614434008897677,
47
+ "miss_rate_1000m": 0.8758032624814631,
48
+ "miss_rate_2000m": 0.7698961937716263
49
+ },
50
+ "extrap_delta_300": {
51
+ "ade_m": 4080.863037109375,
52
+ "fde_m": 10524.166015625,
53
+ "fde_median_m": 9125.39453125,
54
+ "nll_xy_z": -1.427000880241394,
55
+ "coverage_95": 0.439693524468611,
56
+ "n": 8092,
57
+ "miss_rate_500m": 0.9584775086505191,
58
+ "miss_rate_1000m": 0.8755561047948591,
59
+ "miss_rate_2000m": 0.7738507167572911
60
+ }
61
+ },
62
+ "args": {
63
+ "airport": "RKSIa",
64
+ "data_dir": "data",
65
+ "tag": "v6-lam0p1-s2",
66
+ "out_dir": "runs",
67
+ "epochs": 60,
68
+ "batch_size": 64,
69
+ "lr": 0.0001,
70
+ "weight_decay": 0.0001,
71
+ "past_max": 256,
72
+ "past_min": 60,
73
+ "delta_min": 30,
74
+ "delta_max": 120,
75
+ "extrap_delta_max": 300,
76
+ "epoch_multiplier": 1,
77
+ "lambda_jepa": 0.1,
78
+ "ema_decay": 0.998,
79
+ "beta_nll": 0.5,
80
+ "ss_max": 0.0,
81
+ "ss_warmup_frac": 0.5,
82
+ "decoder_mode": "parallel",
83
+ "d_model": 256,
84
+ "n_layers": 4,
85
+ "n_heads": 8,
86
+ "patch_size": 8,
87
+ "seed": 2,
88
+ "num_workers": 2,
89
+ "push_to_hub": true,
90
+ "hub_model_id": "guychuk/flight-jepa-v2",
91
+ "trackio_name": "v6-lam0p1-s2"
92
+ }
93
+ }