Upload v8-pre-hoRKSIa-s0_summary.json with huggingface_hub

v8-pre-hoRKSIa-s0_summary.json

@@ -0,0 +1,101 @@
+{
+  "tag": "v8-pre-hoRKSIa-s0",
+  "lambda_jepa": 0.0,
+  "beta_nll": 0.5,
+  "n_params": 8034446,
+  "best_fde_m": 2497.855712890625,
+  "final": {
+    "in_distribution": {
+      "ade_m": 1405.97900390625,
+      "fde_m": 2497.855712890625,
+      "fde_median_m": 1846.036865234375,
+      "nll_xy_z": -9.78461742401123,
+      "coverage_95": 0.8635689569945625,
+      "n": 8092,
+      "miss_rate_500m": 0.9373455264458724,
+      "miss_rate_1000m": 0.7842313395946614,
+      "miss_rate_2000m": 0.46094908551655955,
+      "per_bucket": {
+        "delta_30_60": {
+          "ade_m": 960.067138671875,
+          "fde_m": 1395.2906494140625,
+          "coverage_95": 0.9182835820895522,
+          "n": 2680
+        },
+        "delta_60_90": {
+          "ade_m": 1357.036376953125,
+          "fde_m": 2411.34716796875,
+          "coverage_95": 0.8526722925457103,
+          "n": 2844
+        },
+        "delta_90_120": {
+          "ade_m": 1888.025146484375,
+          "fde_m": 3659.147705078125,
+          "coverage_95": 0.8217280349981771,
+          "n": 2743
+        }
+      }
+    },
+    "extrap_delta_180": {
+      "ade_m": 2726.699462890625,
+      "fde_m": 6684.978515625,
+      "fde_median_m": 6173.8408203125,
+      "nll_xy_z": -6.818624019622803,
+      "coverage_95": 0.6478002965892239,
+      "n": 8092,
+      "miss_rate_500m": 0.9824518042511122,
+      "miss_rate_1000m": 0.9247404844290658,
+      "miss_rate_2000m": 0.800790904597133
+    },
+    "extrap_delta_300": {
+      "ade_m": 3678.465087890625,
+      "fde_m": 9166.7841796875,
+      "fde_median_m": 7306.09130859375,
+      "nll_xy_z": -2.984591484069824,
+      "coverage_95": 0.4893722194760257,
+      "n": 8092,
+      "miss_rate_500m": 0.9796094908551656,
+      "miss_rate_1000m": 0.9207859614434009,
+      "miss_rate_2000m": 0.8017795353435492
+    }
+  },
+  "args": {
+    "airport": "RKSIa",
+    "multi_airport": "RKSId,ESSA,LSZH",
+    "eval_airport": "RKSIa",
+    "use_airport_token": true,
+    "data_dir": "data",
+    "tag": "v8-pre-hoRKSIa-s0",
+    "out_dir": "runs",
+    "epochs": 60,
+    "batch_size": 64,
+    "lr": 0.0001,
+    "weight_decay": 0.0001,
+    "past_max": 256,
+    "past_min": 60,
+    "delta_min": 30,
+    "delta_max": 120,
+    "extrap_delta_max": 300,
+    "epoch_multiplier": 1,
+    "lambda_jepa": 0.0,
+    "ema_decay": 0.998,
+    "beta_nll": 0.5,
+    "ss_max": 0.0,
+    "ss_warmup_frac": 0.5,
+    "decoder_mode": "parallel",
+    "d_model": 256,
+    "n_layers": 4,
+    "n_heads": 8,
+    "patch_size": 8,
+    "seed": 0,
+    "num_workers": 2,
+    "push_to_hub": true,
+    "hub_model_id": "guychuk/flight-jepa-v2",
+    "pretrained_encoder": "guychuk/flight-jepa-v2",
+    "pretrained_encoder_file": "v8-pretrain-hoRKSIa.pt",
+    "freeze_encoder": false,
+    "held_out_classes": null,
+    "keep_only_classes": null,
+    "trackio_name": "v8-pre-hoRKSIa-s0"
+  }
+}