Upload SIGMA_v13 checkpoints and results

checkpoints_a3_best/best_meta.json

@@ -0,0 +1,7 @@
+{
+  "epoch": 10,
+  "step": 3020,
+  "mrr": 0.9637941398196868,
+  "r1": 0.946,
+  "r10": 0.996
+}

checkpoints_a3_best/best_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00f8ac69bda9ba7d11bb8d30563a93415933a85f958391df872244fb45f86da3
+size 58984949

results_a3/a3_final.json

@@ -0,0 +1,153 @@
+{
+  "config": {
+    "dim": 256,
+    "n_steps": 2,
+    "top_k": 16,
+    "rank": 4,
+    "n_heads": 2,
+    "warmup_ratio": 0.1,
+    "ema_decay": 0.999,
+    "batch_size": 128
+  },
+  "final_metrics": {
+    "mrr": 0.971924885660479,
+    "r1": 0.958,
+    "r5": 0.99,
+    "r10": 0.998
+  },
+  "best_mrr": 0.9637941398196868,
+  "history": {
+    "step": [
+      151,
+      302,
+      453,
+      604,
+      755,
+      906,
+      1057,
+      1208,
+      1359,
+      1510,
+      1661,
+      1812,
+      1963,
+      2114,
+      2265,
+      2416,
+      2567,
+      2718,
+      2869,
+      3020
+    ],
+    "loss": [
+      4.18417600764344,
+      3.4565066201797383,
+      1.407371061921909,
+      1.094078879087966,
+      0.3985431365027333,
+      0.3693263522836546,
+      0.20021145346741012,
+      0.1884872525288964,
+      0.1103784946219021,
+      0.10422923401096798,
+      0.06833229100467353,
+      0.06716461970612703,
+      0.05489896299527181,
+      0.054119892233353575,
+      0.04856092991023664,
+      0.048116377746032564,
+      0.04458457325270634,
+      0.042907547224189664,
+      0.04156933941983229,
+      0.04068860187756502
+    ],
+    "mrr": [
+      0.011174324410224933,
+      0.015715928576686647,
+      0.01904165526061832,
+      0.026162325605719954,
+      0.10163972497412811,
+      0.28441519777968016,
+      0.5058603058619553,
+      0.7110317882665718,
+      0.8148222379150314,
+      0.865726731550368,
+      0.8932233508158076,
+      0.9137322042842024,
+      0.9355255315145559,
+      0.9403312601613473,
+      0.9484149910255737,
+      0.9540925315425316,
+      0.9555926131339925,
+      0.9586867965367964,
+      0.9618191602099245,
+      0.9637941398196868
+    ],
+    "r1": [
+      0.002,
+      0.004,
+      0.006,
+      0.006,
+      0.048,
+      0.178,
+      0.374,
+      0.616,
+      0.742,
+      0.812,
+      0.848,
+      0.874,
+      0.91,
+      0.914,
+      0.928,
+      0.934,
+      0.934,
+      0.938,
+      0.944,
+      0.946
+    ],
+    "grad_norm": [
+      4.954794731140137,
+      6.383665900230408,
+      4.369035260677338,
+      2.579656716585159,
+      1.6275103175640107,
+      1.3361677485704422,
+      0.859700264930725,
+      0.7522292715311051,
+      0.4702634619176388,
+      0.45671847328543663,
+      0.28335013419389726,
+      0.2868455497920513,
+      0.20585744976997375,
+      0.20741430923342705,
+      0.17761142425239085,
+      0.17609935820102693,
+      0.16816730216145515,
+      0.15923061683773995,
+      0.15379879228770732,
+      0.1542398615926504
+    ],
+    "lr": [
+      0.0005,
+      0.001,
+      0.000992403876506104,
+      0.0009698463103929542,
+      0.0009330127018922195,
+      0.000883022221559489,
+      0.0008213938048432696,
+      0.00075,
+      0.0006710100716628344,
+      0.0005868240888334653,
+      0.0005,
+      0.00041317591116653486,
+      0.0003289899283371657,
+      0.0002500000000000001,
+      0.0001786061951567303,
+      0.00011697777844051105,
+      6.698729810778065e-05,
+      3.0153689607045842e-05,
+      7.59612349389599e-06,
+      0.0
+    ]
+  }
+}