Add model metadata from checkpoints2

models/test_indices.json

@@ -0,0 +1,770 @@
+{
+  "train_idx": [
+    86,
+    87,
+    721,
+    1646,
+    1194,
+    1904,
+    602,
+    2000,
+    704,
+    1249,
+    430,
+    2823,
+    2824,
+    845,
+    856,
+    858,
+    861,
+    267,
+    2426,
+    2427,
+    2737,
+    1602,
+    2104,
+    2751,
+    2516,
+    58,
+    2293,
+    1045,
+    2569,
+    2571,
+    2574,
+    2575,
+    2576,
+    2580,
+    2581,
+    2582,
+    2584,
+    2585,
+    2598,
+    2599,
+    2604,
+    2661,
+    676,
+    987,
+    1335,
+    2613,
+    1553,
+    166,
+    364,
+    366,
+    414,
+    415,
+    1514,
+    1550,
+    1950,
+    2039,
+    1186,
+    1419,
+    583,
+    1421,
+    293,
+    860,
+    862,
+    2791,
+    1858,
+    2666,
+    1009,
+    2462,
+    365,
+    1600,
+    424,
+    848,
+    1953,
+    786,
+    1014,
+    2591,
+    2362,
+    2458,
+    243,
+    2456,
+    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,
+    1062,
+    817,
+    2938,
+    2940,
+    2941,
+    2943,
+    1216,
+    2570,
+    2686,
+    2539,
+    1099,
+    1320,
+    1001,
+    1353,
+    1869,
+    1871,
+    1873,
+    1980,
+    2006,
+    2619,
+    2945,
+    2948,
+    2956,
+    876,
+    1201,
+    1204,
+    1206,
+    1208,
+    1212,
+    1213,
+    1215,
+    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,
+    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,
+    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,
+    1664,
+    2022,
+    1800,
+    2176,
+    2492,
+    730,
+    1991,
+    790,
+    1668,
+    1679,
+    271,
+    90,
+    2495,
+    198,
+    1824,
+    1827,
+    1832,
+    1834,
+    1846,
+    1854,
+    2801,
+    89,
+    463,
+    505,
+    506,
+    508,
+    509,
+    510,
+    849,
+    1163,
+    1250,
+    1251,
+    1252,
+    1253,
+    1263,
+    1721,
+    2156,
+    2182,
+    2216,
+    2223,
+    2475,
+    2629,
+    2671,
+    2693,
+    2695,
+    2697,
+    2700,
+    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,
+    2514,
+    1474,
+    2142,
+    66,
+    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,
+    2633,
+    747,
+    748,
+    172,
+    173,
+    183,
+    1341,
+    1390,
+    1393,
+    1203,
+    1205,
+    1207,
+    1209,
+    1211,
+    404,
+    2500,
+    1988,
+    2869,
+    2647,
+    2910,
+    2911,
+    2912,
+    954,
+    1330,
+    1333,
+    492,
+    1345,
+    1346,
+    1019,
+    2825,
+    2827,
+    2830,
+    2833,
+    2836,
+    2837,
+    1102,
+    2796,
+    2799,
+    403,
+    2393,
+    731,
+    1491,
+    882,
+    1042,
+    826,
+    399,
+    2507,
+    1893,
+    2194,
+    910,
+    2092,
+    2089,
+    1401,
+    2979,
+    2982,
+    2983,
+    2984,
+    2985,
+    2990,
+    2993,
+    2997,
+    2998,
+    859,
+    2897,
+    2616,
+    2370,
+    2373,
+    2597,
+    716,
+    2124,
+    2125,
+    2126,
+    2128,
+    2129,
+    2130,
+    2131,
+    2314,
+    2315,
+    2316,
+    2420,
+    2778,
+    1056,
+    1057,
+    746,
+    749,
+    752,
+    2684,
+    1975,
+    2141,
+    2135,
+    878,
+    2401,
+    2404,
+    2406,
+    2308,
+    2708,
+    1942,
+    1944,
+    1630,
+    376,
+    377,
+    378,
+    1856,
+    1507,
+    1509,
+    2341,
+    2309
+  ],
+  "val_idx": [
+    2276,
+    1650,
+    131,
+    294,
+    295,
+    296,
+    298,
+    496,
+    499,
+    2160,
+    2168,
+    2170,
+    346,
+    905,
+    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,
+    1926,
+    2936,
+    2937,
+    2685,
+    1061,
+    1064,
+    1120,
+    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,
+    2853,
+    2855,
+    1726,
+    52,
+    68,
+    69,
+    874,
+    1174,
+    2658,
+    1645,
+    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,
+    713,
+    1104,
+    559,
+    1609,
+    2066,
+    537,
+    148,
+    902,
+    2758,
+    765,
+    2561,
+    117,
+    118,
+    1161,
+    1265,
+    1214,
+    1223,
+    38,
+    39,
+    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,
+    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,
+    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
+  ],
+  "args": {
+    "anno": "samples.jsonl",
+    "backbone": "ast",
+    "ast_model": "MIT/ast-finetuned-audioset-10-10-0.4593",
+    "unfreeze": 0,
+    "unfreeze_all": true,
+    "epochs": 10,
+    "batch_size": 16,
+    "lr": 0.0001,
+    "weight_decay": 0.0001,
+    "num_slices": 10,
+    "bars": 4,
+    "n_mels": 128,
+    "sample_rate": 16000,
+    "neg_prob": 0.5,
+    "augment": true,
+    "cached_only": true,
+    "accum_steps": 1,
+    "val_frac": 0.15,
+    "test_frac": 0.15,
+    "seed": 42,
+    "ckpt_dir": "checkpoints2",
+    "workers": 0,
+    "pos_weight": 1.5,
+    "artist_split": true,
+    "stage": 3,
+    "mask_weight": 1.0,
+    "resume": "checkpoints2/best.pt"
+  }
+}