Update checkpoints_all32_d128_block_reserve_0_1_2_35/search_step_1000.compare_retrieval.json
Browse files
checkpoints_all32_d128_block_reserve_0_1_2_35/search_step_1000.compare_retrieval.json
ADDED
|
@@ -0,0 +1,159 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model": "Qwen/Qwen3-4B-Instruct-2507",
|
| 3 |
+
"ckpt": "/tmp/checkpoints_all32_d128_block_reserve_0_1_2_35/search_step_1000.pt",
|
| 4 |
+
"by_K": {
|
| 5 |
+
"128": {
|
| 6 |
+
"raw_qk": {
|
| 7 |
+
"per_layer": {
|
| 8 |
+
"3": 0.9387637376785278,
|
| 9 |
+
"4": 0.9440614581108093,
|
| 10 |
+
"5": 0.96429443359375,
|
| 11 |
+
"6": 0.9560344219207764,
|
| 12 |
+
"7": 0.9818542897701263,
|
| 13 |
+
"8": 0.9707900285720825,
|
| 14 |
+
"9": 0.9587394297122955,
|
| 15 |
+
"10": 0.9742867350578308,
|
| 16 |
+
"11": 0.9763059914112091,
|
| 17 |
+
"12": 0.9611693322658539,
|
| 18 |
+
"13": 0.9707654714584351,
|
| 19 |
+
"14": 0.9733296930789948,
|
| 20 |
+
"15": 0.9675203263759613,
|
| 21 |
+
"16": 0.9556995928287506,
|
| 22 |
+
"17": 0.9593449831008911,
|
| 23 |
+
"18": 0.9652450382709503,
|
| 24 |
+
"19": 0.9612957835197449,
|
| 25 |
+
"20": 0.9591256380081177,
|
| 26 |
+
"21": 0.9657162129878998,
|
| 27 |
+
"22": 0.962736189365387,
|
| 28 |
+
"23": 0.9788879752159119,
|
| 29 |
+
"24": 0.9710511267185211,
|
| 30 |
+
"25": 0.9860879778862,
|
| 31 |
+
"26": 0.9775076508522034,
|
| 32 |
+
"27": 0.977691113948822,
|
| 33 |
+
"28": 0.9794201850891113,
|
| 34 |
+
"29": 0.9821525514125824,
|
| 35 |
+
"30": 0.9875719547271729,
|
| 36 |
+
"31": 0.9842455685138702,
|
| 37 |
+
"32": 0.9791028797626495,
|
| 38 |
+
"33": 0.9772593379020691,
|
| 39 |
+
"34": 0.9759000241756439
|
| 40 |
+
},
|
| 41 |
+
"avg": 0.969498660415411
|
| 42 |
+
},
|
| 43 |
+
"learned": {
|
| 44 |
+
"per_layer": {
|
| 45 |
+
"3": 0.9244755506515503,
|
| 46 |
+
"4": 0.9368151426315308,
|
| 47 |
+
"5": 0.9484244585037231,
|
| 48 |
+
"6": 0.939052551984787,
|
| 49 |
+
"7": 0.9830892086029053,
|
| 50 |
+
"8": 0.9710568785667419,
|
| 51 |
+
"9": 0.9766820967197418,
|
| 52 |
+
"10": 0.9707906544208527,
|
| 53 |
+
"11": 0.9762083292007446,
|
| 54 |
+
"12": 0.9698493182659149,
|
| 55 |
+
"13": 0.9710436761379242,
|
| 56 |
+
"14": 0.9732040166854858,
|
| 57 |
+
"15": 0.972749650478363,
|
| 58 |
+
"16": 0.9609748125076294,
|
| 59 |
+
"17": 0.9665586054325104,
|
| 60 |
+
"18": 0.9724838137626648,
|
| 61 |
+
"19": 0.9691765010356903,
|
| 62 |
+
"20": 0.9757413268089294,
|
| 63 |
+
"21": 0.9799380302429199,
|
| 64 |
+
"22": 0.9698161482810974,
|
| 65 |
+
"23": 0.9849565327167511,
|
| 66 |
+
"24": 0.9785935282707214,
|
| 67 |
+
"25": 0.9885723888874054,
|
| 68 |
+
"26": 0.9855012595653534,
|
| 69 |
+
"27": 0.9834729731082916,
|
| 70 |
+
"28": 0.9848337769508362,
|
| 71 |
+
"29": 0.98735311627388,
|
| 72 |
+
"30": 0.9873296022415161,
|
| 73 |
+
"31": 0.9829616844654083,
|
| 74 |
+
"32": 0.9795172214508057,
|
| 75 |
+
"33": 0.9666652083396912,
|
| 76 |
+
"34": 0.9603306651115417
|
| 77 |
+
},
|
| 78 |
+
"avg": 0.9711943352594972
|
| 79 |
+
}
|
| 80 |
+
},
|
| 81 |
+
"256": {
|
| 82 |
+
"raw_qk": {
|
| 83 |
+
"per_layer": {
|
| 84 |
+
"3": 0.9856403172016144,
|
| 85 |
+
"4": 0.9863723814487457,
|
| 86 |
+
"5": 0.9927175343036652,
|
| 87 |
+
"6": 0.9899796843528748,
|
| 88 |
+
"7": 0.9961064755916595,
|
| 89 |
+
"8": 0.9944310188293457,
|
| 90 |
+
"9": 0.9924624264240265,
|
| 91 |
+
"10": 0.9945598244667053,
|
| 92 |
+
"11": 0.9956617951393127,
|
| 93 |
+
"12": 0.9914700388908386,
|
| 94 |
+
"13": 0.9947893917560577,
|
| 95 |
+
"14": 0.9956192076206207,
|
| 96 |
+
"15": 0.9947386384010315,
|
| 97 |
+
"16": 0.9918601512908936,
|
| 98 |
+
"17": 0.9932802319526672,
|
| 99 |
+
"18": 0.994801938533783,
|
| 100 |
+
"19": 0.9929636716842651,
|
| 101 |
+
"20": 0.9928486049175262,
|
| 102 |
+
"21": 0.9940518736839294,
|
| 103 |
+
"22": 0.9925838708877563,
|
| 104 |
+
"23": 0.9956205189228058,
|
| 105 |
+
"24": 0.9944405257701874,
|
| 106 |
+
"25": 0.99702388048172,
|
| 107 |
+
"26": 0.9963724911212921,
|
| 108 |
+
"27": 0.9952751994132996,
|
| 109 |
+
"28": 0.9952269494533539,
|
| 110 |
+
"29": 0.9970046579837799,
|
| 111 |
+
"30": 0.9976439476013184,
|
| 112 |
+
"31": 0.9967932999134064,
|
| 113 |
+
"32": 0.9949435591697693,
|
| 114 |
+
"33": 0.9954463541507721,
|
| 115 |
+
"34": 0.9949491024017334
|
| 116 |
+
},
|
| 117 |
+
"avg": 0.9939899863675237
|
| 118 |
+
},
|
| 119 |
+
"learned": {
|
| 120 |
+
"per_layer": {
|
| 121 |
+
"3": 0.9771333336830139,
|
| 122 |
+
"4": 0.9823162257671356,
|
| 123 |
+
"5": 0.9861737489700317,
|
| 124 |
+
"6": 0.9809035956859589,
|
| 125 |
+
"7": 0.9963934123516083,
|
| 126 |
+
"8": 0.9932794272899628,
|
| 127 |
+
"9": 0.9953930974006653,
|
| 128 |
+
"10": 0.9923880398273468,
|
| 129 |
+
"11": 0.994959145784378,
|
| 130 |
+
"12": 0.9928728640079498,
|
| 131 |
+
"13": 0.9931342303752899,
|
| 132 |
+
"14": 0.9937034547328949,
|
| 133 |
+
"15": 0.9947409331798553,
|
| 134 |
+
"16": 0.9912147521972656,
|
| 135 |
+
"17": 0.99372997879982,
|
| 136 |
+
"18": 0.9954918026924133,
|
| 137 |
+
"19": 0.9936331808567047,
|
| 138 |
+
"20": 0.9959999024868011,
|
| 139 |
+
"21": 0.99661585688591,
|
| 140 |
+
"22": 0.993903785943985,
|
| 141 |
+
"23": 0.9974819421768188,
|
| 142 |
+
"24": 0.9961839318275452,
|
| 143 |
+
"25": 0.9976893067359924,
|
| 144 |
+
"26": 0.9971556663513184,
|
| 145 |
+
"27": 0.9965596199035645,
|
| 146 |
+
"28": 0.9969134330749512,
|
| 147 |
+
"29": 0.9974974989891052,
|
| 148 |
+
"30": 0.9971778690814972,
|
| 149 |
+
"31": 0.9964099228382111,
|
| 150 |
+
"32": 0.9948908686637878,
|
| 151 |
+
"33": 0.9909328520298004,
|
| 152 |
+
"34": 0.98967644572258
|
| 153 |
+
},
|
| 154 |
+
"avg": 0.9932046914473176
|
| 155 |
+
}
|
| 156 |
+
}
|
| 157 |
+
},
|
| 158 |
+
"learned_over_raw_K128": 1.0017490223693137
|
| 159 |
+
}
|