Update test_indices.json from checkpoints3

models/test_indices.json

@@ -1,33 +1,111 @@
 {
   "train_idx": [
-    86,
-    87,
-    721,
-    1646,
-    1194,
-    1904,
-    602,
-    2000,
-    704,
-    1249,
+    559,
+    1609,
+    1099,
+    1320,
+    1893,
+    1660,
+    1661,
+    2075,
+    2073,
+    2341,
+    716,
+    166,
+    364,
+    366,
+    414,
+    415,
+    2500,
+    2157,
+    1548,
+    1565,
+    2819,
+    1340,
+    1429,
+    1430,
+    1432,
+    1433,
+    1434,
+    1435,
+    1436,
+    1437,
+    1439,
+    1440,
+    1442,
+    1443,
+    1444,
+    1445,
+    1446,
+    1448,
+    1450,
+    1451,
+    1452,
+    1453,
+    1456,
+    1526,
+    1527,
+    2573,
+    299,
+    300,
+    358,
+    361,
+    362,
+    847,
+    817,
+    2706,
+    322,
+    2176,
+    2597,
+    2020,
+    2455,
+    2457,
+    806,
+    2104,
+    1316,
+    786,
+    1014,
+    1518,
     430,
-    2823,
-    2824,
-    845,
-    856,
-    858,
-    861,
+    1613,
+    1615,
+    1616,
+    1618,
+    1619,
+    1621,
+    1622,
+    1623,
+    1624,
+    1625,
+    1626,
+    1919,
+    1925,
+    1929,
+    1645,
+    2825,
+    2827,
+    2830,
+    2833,
+    2836,
+    2837,
+    2838,
     267,
     2426,
     2427,
     2737,
-    1602,
-    2104,
+    814,
+    815,
+    816,
+    1224,
+    1920,
+    1975,
+    2141,
     2751,
-    2516,
-    58,
+    2765,
+    2509,
     2293,
-    1045,
+    977,
+    978,
     2569,
     2571,
     2574,
@@ -41,97 +119,157 @@
     2598,
     2599,
     2604,
+    2605,
     2661,
-    676,
-    987,
-    1335,
-    2613,
+    2124,
+    2125,
+    2126,
+    2128,
+    2129,
+    2130,
+    2131,
+    2314,
+    2315,
+    2316,
+    2420,
+    2778,
+    1216,
+    2570,
+    2686,
+    2868,
+    492,
+    1794,
+    663,
+    1904,
+    566,
+    2708,
+    360,
+    409,
+    410,
+    411,
     1553,
-    166,
-    364,
-    366,
-    414,
-    415,
-    1514,
-    1550,
-    1950,
-    2039,
-    1186,
-    1419,
-    583,
-    1421,
-    293,
-    860,
-    862,
+    1736,
+    404,
     2791,
-    1858,
-    2666,
+    730,
+    1667,
     1009,
-    2462,
+    2883,
     365,
-    1600,
+    1650,
     424,
     848,
-    1953,
-    786,
-    1014,
+    1856,
+    2775,
+    2776,
+    2777,
     2591,
-    2362,
-    2458,
-    243,
-    2456,
+    2889,
+    148,
+    1395,
+    1722,
     2579,
-    543,
-    544,
-    545,
-    546,
-    547,
-    548,
-    550,
-    551,
-    2765,
-    104,
-    2101,
-    2503,
-    779,
-    1644,
-    2558,
-    2562,
-    2975,
-    759,
-    603,
-    782,
-    1124,
-    1548,
-    1565,
-    2819,
-    385,
-    386,
-    360,
+    271,
+    2458,
+    574,
+    576,
+    582,
+    585,
+    2754,
+    131,
+    777,
+    778,
+    2266,
+    117,
+    118,
+    1104,
+    2279,
+    2281,
+    1390,
+    1393,
+    2587,
+    2012,
+    765,
+    910,
     1062,
-    817,
-    2938,
-    2940,
-    2941,
-    2943,
-    1216,
-    2570,
-    2686,
-    2539,
-    1099,
-    1320,
-    1001,
-    1353,
-    1869,
-    1871,
-    1873,
-    1980,
-    2006,
-    2619,
-    2945,
-    2948,
-    2956,
+    376,
+    377,
+    378,
+    1377,
+    321,
+    324,
+    325,
+    327,
+    333,
+    336,
+    337,
+    339,
+    340,
+    341,
+    342,
+    1600,
+    2865,
+    537,
+    504,
     876,
+    1879,
+    779,
+    2881,
+    1942,
+    1944,
+    1656,
+    2360,
+    2008,
+    2666,
+    1,
+    2,
+    1875,
+    1877,
+    602,
+    1341,
+    1897,
+    1161,
+    1493,
+    1421,
+    855,
+    553,
+    2142,
+    1641,
+    2362,
+    2309,
+    2495,
+    89,
+    532,
+    534,
+    1663,
+    1665,
+    2977,
+    293,
+    2502,
+    2255,
+    2257,
+    2260,
+    2263,
+    2270,
+    2274,
+    2319,
+    2322,
+    66,
+    1194,
+    2039,
+    782,
+    621,
+    2085,
+    1214,
+    1223,
+    2062,
+    363,
+    2542,
+    2544,
+    2545,
+    1338,
+    1071,
+    596,
     1201,
     1204,
     1206,
@@ -142,151 +280,57 @@
     1218,
     1220,
     1222,
-    460,
-    461,
-    464,
-    465,
-    558,
-    565,
-    570,
-    573,
-    622,
-    625,
-    627,
-    629,
-    632,
-    633,
-    634,
-    635,
-    636,
-    641,
-    2886,
-    2888,
-    1171,
-    1172,
-    1177,
-    2019,
-    2883,
-    855,
-    1736,
-    1528,
-    2587,
-    1299,
-    922,
-    1493,
+    1419,
+    1957,
+    172,
+    173,
+    174,
+    183,
+    38,
+    39,
+    882,
+    2371,
+    1938,
+    1939,
     219,
-    739,
-    553,
-    1210,
-    2865,
-    1556,
-    1438,
-    1753,
-    839,
-    2921,
-    2922,
-    2085,
-    2183,
-    2654,
-    1415,
-    228,
-    241,
-    242,
-    1879,
-    1359,
-    1363,
-    2266,
-    1936,
-    1937,
-    458,
-    2062,
-    977,
-    978,
-    1663,
-    1665,
-    2977,
-    596,
     1859,
-    1071,
-    1920,
-    2313,
-    2450,
-    1794,
-    2594,
-    957,
-    959,
-    960,
-    2808,
-    2810,
-    416,
-    648,
-    649,
-    651,
-    653,
-    656,
-    750,
-    751,
-    806,
+    305,
+    348,
+    349,
+    351,
+    353,
+    354,
+    355,
+    2355,
+    2135,
+    346,
+    440,
+    2101,
+    1045,
+    2654,
     2840,
     2841,
-    471,
-    2020,
-    2868,
-    1138,
-    1940,
-    1428,
-    1982,
-    1613,
-    1615,
-    1616,
-    1618,
-    1619,
-    1621,
-    1622,
-    1623,
-    1624,
-    1625,
-    1626,
-    1919,
-    1925,
-    1929,
-    410,
-    411,
-    440,
-    2224,
-    2255,
-    2257,
-    2260,
-    2263,
-    2270,
-    2274,
-    2319,
-    2322,
-    1897,
-    567,
-    1617,
+    928,
     1664,
-    2022,
-    1800,
-    2176,
-    2492,
-    730,
+    1335,
+    1993,
+    1331,
+    856,
+    858,
+    861,
+    2761,
+    2766,
+    2774,
     1991,
-    790,
-    1668,
-    1679,
-    271,
-    90,
-    2495,
-    198,
-    1824,
-    1827,
-    1832,
-    1834,
-    1846,
-    1854,
-    2801,
-    89,
+    1359,
+    1360,
+    1363,
+    1602,
+    2000,
+    1841,
+    1842,
+    1753,
+    1457,
     463,
     505,
     506,
@@ -315,428 +359,404 @@
     2701,
     2702,
     2716,
-    2008,
-    1075,
-    371,
-    372,
-    793,
-    795,
-    796,
-    797,
-    798,
-    799,
-    800,
-    802,
-    803,
-    1551,
-    1552,
-    1563,
-    1564,
-    1566,
-    1567,
-    2719,
-    2720,
-    2721,
-    2723,
-    2725,
-    2738,
-    2739,
-    2752,
-    2753,
-    2784,
-    2996,
-    2999,
-    1641,
-    2531,
-    566,
-    1,
-    2,
-    1875,
-    1877,
+    1982,
     2514,
-    1474,
-    2142,
-    66,
+    747,
+    748,
+    1283,
+    2219,
+    2261,
+    2262,
+    2269,
+    1042,
+    402,
+    2105,
+    731,
+    1491,
+    1304,
+    460,
+    461,
+    464,
+    465,
+    1514,
+    1550,
+    1950,
     1508,
     1511,
     1525,
     1531,
     1532,
-    669,
-    2728,
-    2732,
-    2881,
-    363,
-    2542,
-    2544,
-    2545,
-    574,
-    576,
-    582,
-    585,
-    2396,
-    2399,
-    2400,
-    2402,
-    2403,
-    2405,
-    2407,
-    2409,
-    2410,
-    2412,
-    2417,
-    2418,
-    2421,
-    2422,
-    2423,
-    2428,
-    2433,
-    2435,
-    2436,
-    2439,
-    2441,
-    2442,
-    2443,
+    1345,
+    1346,
+    2823,
+    2824,
+    113,
+    471,
+    2092,
     2633,
-    747,
-    748,
-    172,
-    173,
-    183,
-    1341,
-    1390,
-    1393,
-    1203,
-    1205,
-    1207,
-    1209,
-    1211,
-    404,
-    2500,
-    1988,
+    1055,
+    294,
+    295,
+    296,
+    298,
+    496,
+    499,
+    2160,
+    2168,
+    2170,
+    859,
+    2897,
+    1186,
+    1953,
     2869,
-    2647,
+    1927,
+    1474,
+    1726,
+    1936,
+    1937,
+    1171,
+    1172,
+    1175,
+    1177,
+    2019,
     2910,
     2911,
     2912,
-    954,
+    514,
+    2194,
+    2994,
     1330,
     1333,
-    492,
-    1345,
-    1346,
-    1019,
-    2825,
-    2827,
-    2830,
-    2833,
-    2836,
-    2837,
+    1056,
+    1057,
+    845,
+    558,
+    565,
+    570,
+    573,
+    622,
+    625,
+    627,
+    629,
+    632,
+    633,
+    634,
+    635,
+    636,
+    641,
     1102,
-    2796,
-    2799,
-    403,
-    2393,
-    731,
-    1491,
-    882,
-    1042,
-    826,
-    399,
+    1174,
+    2685,
+    2886,
+    2888,
+    1306,
+    1952,
+    2492,
+    168,
     2507,
-    1893,
-    2194,
-    910,
-    2092,
-    2089,
+    2022,
+    2921,
+    2922,
+    583,
+    1617,
+    1546,
     1401,
-    2979,
-    2982,
-    2983,
-    2984,
-    2985,
-    2990,
-    2993,
-    2997,
-    2998,
-    859,
-    2897,
+    2503,
+    839,
+    2308,
     2616,
-    2370,
-    2373,
-    2597,
-    716,
-    2124,
-    2125,
-    2126,
-    2128,
-    2129,
-    2130,
-    2131,
-    2314,
-    2315,
-    2316,
-    2420,
-    2778,
-    1056,
-    1057,
+    543,
+    544,
+    545,
+    546,
+    547,
+    548,
+    550,
+    551,
+    2589,
+    2590,
+    2615,
+    2638,
+    2644,
+    2648,
+    676,
+    987,
+    2531,
+    2938,
+    2940,
+    2941,
+    2943,
     746,
     749,
     752,
     2684,
-    1975,
-    2141,
-    2135,
-    878,
-    2401,
-    2404,
-    2406,
-    2308,
-    2708,
-    1942,
-    1944,
-    1630,
-    376,
-    377,
-    378,
-    1856,
+    2559,
+    1759,
+    860,
+    862,
+    243,
+    2102,
+    2289,
+    603,
+    1528,
+    403,
+    2393,
+    1858,
+    954,
     1507,
     1509,
-    2341,
-    2309
+    698,
+    156,
+    1100
   ],
   "val_idx": [
+    119,
+    370,
+    1019,
+    957,
+    959,
+    960,
+    1272,
+    1274,
+    1300,
+    1302,
+    1303,
+    874,
+    1974,
+    1701,
     2276,
-    1650,
-    131,
-    294,
-    295,
-    296,
-    298,
-    496,
-    499,
-    2160,
-    2168,
-    2170,
-    346,
-    905,
+    2613,
+    2715,
+    1124,
+    2730,
+    2183,
+    2450,
+    2979,
+    2982,
+    2983,
+    2984,
+    2985,
+    2990,
+    2993,
+    2997,
+    2998,
+    416,
+    648,
+    649,
+    651,
+    653,
+    656,
+    750,
+    751,
+    2505,
+    2895,
+    2891,
+    458,
+    790,
+    1668,
+    1679,
+    902,
+    2758,
+    2468,
+    2469,
+    2470,
+    2471,
+    2472,
+    2473,
+    2476,
+    2477,
+    2478,
+    2479,
+    1203,
+    1205,
+    1207,
+    1209,
+    1211,
+    2390,
+    2391,
+    1299,
+    2224,
+    1428,
+    1940,
+    90,
     702,
-    2075,
-    1510,
-    2049,
-    1656,
-    2360,
-    2502,
-    1526,
-    1527,
-    1331,
-    2885,
-    1957,
-    1355,
-    2761,
-    2766,
-    2774,
-    1340,
-    1429,
-    1430,
-    1432,
-    1433,
-    1434,
-    1435,
-    1436,
-    1437,
-    1439,
-    1440,
-    1442,
-    1443,
-    1444,
-    1445,
-    1446,
-    1448,
-    1450,
-    1451,
-    1452,
-    1453,
-    1456,
-    1377,
+    198,
+    1821,
+    1824,
+    1827,
+    1832,
+    1834,
+    1846,
+    1854,
+    2801,
+    1584,
     1926,
-    2936,
-    2937,
-    2685,
+    2647,
+    2456,
+    1334,
+    2924,
     1061,
     1064,
     1120,
+    1264,
     1266,
     1292,
-    1683,
-    1701,
-    1660,
-    1661,
-    532,
-    534,
-    2889,
-    1395,
-    1228,
-    431,
-    432,
-    433,
-    434,
-    698,
-    2924,
-    2520,
-    814,
-    815,
-    816,
-    1224,
-    2573,
-    299,
-    300,
-    358,
-    361,
-    362,
-    2219,
-    2261,
-    2262,
-    2269,
-    1584,
-    1845,
+    2558,
+    2562,
+    2975,
+    2416,
     2853,
     2855,
-    1726,
+    1438,
+    1228,
+    1644,
+    2808,
+    2810,
+    104,
+    2936,
+    2937,
+    2396,
+    2399,
+    2400,
+    2402,
+    2403,
+    2405,
+    2407,
+    2409,
+    2410,
+    2412,
+    2417,
+    2418,
+    2421,
+    2422,
+    2423,
+    2428,
+    2433,
+    2435,
+    2436,
+    2439,
+    2441,
+    2442,
+    2443
+  ],
+  "test_idx": [
+    2594,
+    86,
+    87,
+    2561,
+    2462,
+    200,
+    1138,
+    2271,
     52,
     68,
     69,
-    874,
-    1174,
+    567,
+    1845,
+    2386,
+    228,
+    241,
+    242,
+    885,
+    886,
+    826,
+    739,
+    2885,
+    704,
+    1249,
+    1415,
     2658,
-    1645,
+    721,
+    1646,
+    2516,
+    431,
+    432,
+    433,
+    434,
+    1265,
+    58,
+    1988,
+    878,
+    2401,
+    2404,
+    2406,
+    1556,
     2294,
-    321,
-    324,
-    325,
-    327,
-    333,
-    336,
-    337,
-    339,
-    340,
-    341,
-    342
-  ],
-  "test_idx": [
-    1993,
-    2279,
-    2281,
-    2371,
-    928,
-    1938,
-    1939,
-    1952,
-    2355,
-    1518,
-    2455,
-    2457,
+    399,
+    1001,
+    1353,
+    1869,
+    1871,
+    1873,
+    1980,
+    2006,
+    2619,
+    2945,
+    2948,
+    2956,
+    884,
+    669,
+    2728,
+    2732,
     713,
-    1104,
-    559,
-    1609,
-    2066,
-    537,
-    148,
-    902,
-    2758,
-    765,
-    2561,
-    117,
-    118,
-    1161,
-    1265,
-    1214,
-    1223,
-    38,
-    39,
+    753,
+    759,
+    2089,
+    745,
+    1510,
     2762,
-    1272,
-    1274,
-    1300,
-    1302,
-    1303,
-    2775,
-    2776,
-    2777,
-    1546,
-    2157,
-    884,
-    514,
-    370,
-    2715,
-    1334,
-    2012,
-    113,
-    2589,
-    2590,
-    2615,
-    2638,
-    2644,
-    2648,
-    1927,
-    2509,
-    1304,
-    820,
+    1683,
+    2520,
     2560,
-    1100,
-    1974,
-    2706,
-    2754,
-    2994,
-    1338,
-    663,
-    885,
-    886,
-    119,
-    168,
-    2730,
-    2102,
-    2289,
-    1841,
-    1842,
-    1316,
-    2559,
-    2891,
-    402,
-    2105,
-    504,
-    2390,
-    2391,
-    2505,
-    2895,
-    745,
+    905,
+    1355,
+    1210,
+    2066,
+    252,
     406,
-    847,
-    2416,
-    2271,
-    777,
-    778,
-    305,
-    348,
-    349,
-    351,
-    353,
-    354,
-    355,
-    1457,
-    1759,
-    200,
-    2468,
-    2469,
-    2470,
-    2471,
-    2472,
-    2473,
-    2476,
-    2477,
-    2478,
-    2479
+    2539,
+    2049,
+    385,
+    386,
+    1800,
+    2796,
+    2799,
+    1630,
+    922,
+    2313,
+    1075,
+    820,
+    2370,
+    2373,
+    371,
+    372,
+    793,
+    795,
+    796,
+    797,
+    798,
+    799,
+    800,
+    802,
+    803,
+    1551,
+    1552,
+    1563,
+    1564,
+    1566,
+    1567,
+    2719,
+    2720,
+    2721,
+    2723,
+    2725,
+    2738,
+    2739,
+    2752,
+    2753,
+    2784,
+    2996,
+    2999
   ],
   "args": {
     "anno": "samples.jsonl",
@@ -745,7 +765,7 @@
     "unfreeze": 0,
     "unfreeze_all": true,
     "epochs": 10,
-    "batch_size": 16,
+    "batch_size": 24,
     "lr": 0.0001,
     "weight_decay": 0.0001,
     "num_slices": 10,
@@ -759,11 +779,11 @@
     "val_frac": 0.15,
     "test_frac": 0.15,
     "seed": 42,
-    "ckpt_dir": "checkpoints2",
+    "ckpt_dir": "checkpoints3",
     "workers": 0,
     "pos_weight": 1.5,
     "artist_split": true,
-    "stage": 3,
+    "stage": 2,
     "mask_weight": 1.0,
     "resume": "checkpoints2/best.pt"
   }