guychuk commited on
Commit
79c92aa
·
verified ·
1 Parent(s): 05ef834

Upload v5-lam0p1-s2_summary.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. v5-lam0p1-s2_summary.json +92 -0
v5-lam0p1-s2_summary.json ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "tag": "v5-lam0p1-s2",
3
+ "lambda_jepa": 0.1,
4
+ "beta_nll": 0.5,
5
+ "n_params": 7539335,
6
+ "best_fde_m": 2711.35205078125,
7
+ "final": {
8
+ "in_distribution": {
9
+ "ade_m": 1243.3875732421875,
10
+ "fde_m": 2711.35205078125,
11
+ "fde_median_m": 1913.610107421875,
12
+ "nll_xy_z": 233.97055053710938,
13
+ "coverage_95": 0.06265447355412754,
14
+ "n": 8092,
15
+ "miss_rate_500m": 0.9221453287197232,
16
+ "miss_rate_1000m": 0.7399901136925358,
17
+ "miss_rate_2000m": 0.48158675234799797,
18
+ "per_bucket": {
19
+ "delta_30_60": {
20
+ "ade_m": 662.1484985351562,
21
+ "fde_m": 1286.1925048828125,
22
+ "coverage_95": 0.12351885098743268,
23
+ "n": 2785
24
+ },
25
+ "delta_60_90": {
26
+ "ade_m": 1179.477294921875,
27
+ "fde_m": 2581.5615234375,
28
+ "coverage_95": 0.04786262331019364,
29
+ "n": 2737
30
+ },
31
+ "delta_90_120": {
32
+ "ade_m": 1898.5810546875,
33
+ "fde_m": 4289.580078125,
34
+ "coverage_95": 0.014197306152166,
35
+ "n": 2747
36
+ }
37
+ }
38
+ },
39
+ "extrap_delta_180": {
40
+ "ade_m": 5627.31103515625,
41
+ "fde_m": 9092.3388671875,
42
+ "fde_median_m": 7380.802734375,
43
+ "nll_xy_z": 2156.40478515625,
44
+ "coverage_95": 0.0022244191794364805,
45
+ "n": 8092,
46
+ "miss_rate_500m": 1.0,
47
+ "miss_rate_1000m": 1.0,
48
+ "miss_rate_2000m": 0.9972812654473554
49
+ },
50
+ "extrap_delta_300": {
51
+ "ade_m": 10398.8837890625,
52
+ "fde_m": 14401.6572265625,
53
+ "fde_median_m": 11652.9150390625,
54
+ "nll_xy_z": 5440.7744140625,
55
+ "coverage_95": 0.0004943153732081067,
56
+ "n": 8092,
57
+ "miss_rate_500m": 1.0,
58
+ "miss_rate_1000m": 1.0,
59
+ "miss_rate_2000m": 0.999876421156698
60
+ }
61
+ },
62
+ "args": {
63
+ "airport": "RKSIa",
64
+ "data_dir": "data",
65
+ "tag": "v5-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.5,
81
+ "ss_warmup_frac": 0.5,
82
+ "d_model": 256,
83
+ "n_layers": 4,
84
+ "n_heads": 8,
85
+ "patch_size": 8,
86
+ "seed": 2,
87
+ "num_workers": 2,
88
+ "push_to_hub": true,
89
+ "hub_model_id": "guychuk/flight-jepa-v2",
90
+ "trackio_name": "v5-lam0p1-s2"
91
+ }
92
+ }