{ "best_metric": 0.863021420518602, "best_model_checkpoint": "dinov2-base-finetuned-galaxy10-decals/checkpoint-906", "epoch": 29.76, "eval_steps": 500, "global_step": 930, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.32, "learning_rate": 5.376344086021506e-06, "loss": 2.6504, "step": 10 }, { "epoch": 0.64, "learning_rate": 1.0752688172043012e-05, "loss": 1.771, "step": 20 }, { "epoch": 0.96, "learning_rate": 1.6129032258064517e-05, "loss": 1.1698, "step": 30 }, { "epoch": 0.99, "eval_accuracy": 0.6634723788049606, "eval_f1": 0.6488201253523778, "eval_loss": 0.910478949546814, "eval_precision": 0.6861326245634023, "eval_recall": 0.6634723788049606, "eval_runtime": 13.0532, "eval_samples_per_second": 135.905, "eval_steps_per_second": 1.073, "step": 31 }, { "epoch": 1.28, "learning_rate": 2.1505376344086024e-05, "loss": 0.9385, "step": 40 }, { "epoch": 1.6, "learning_rate": 2.6881720430107527e-05, "loss": 0.7928, "step": 50 }, { "epoch": 1.92, "learning_rate": 3.2258064516129034e-05, "loss": 0.7528, "step": 60 }, { "epoch": 1.98, "eval_accuracy": 0.7790304396843292, "eval_f1": 0.7612637488816166, "eval_loss": 0.6493567228317261, "eval_precision": 0.8146646455914968, "eval_recall": 0.7790304396843292, "eval_runtime": 11.0333, "eval_samples_per_second": 160.786, "eval_steps_per_second": 1.269, "step": 62 }, { "epoch": 2.24, "learning_rate": 3.763440860215054e-05, "loss": 0.7283, "step": 70 }, { "epoch": 2.56, "learning_rate": 4.301075268817205e-05, "loss": 0.761, "step": 80 }, { "epoch": 2.88, "learning_rate": 4.8387096774193554e-05, "loss": 0.6893, "step": 90 }, { "epoch": 2.98, "eval_accuracy": 0.794250281848929, "eval_f1": 0.7912121355030378, "eval_loss": 0.6537806391716003, "eval_precision": 0.8019920570293133, "eval_recall": 0.794250281848929, "eval_runtime": 10.9026, "eval_samples_per_second": 162.714, "eval_steps_per_second": 1.284, "step": 93 }, { "epoch": 3.2, "learning_rate": 4.9581839904420555e-05, "loss": 0.7426, "step": 100 }, { "epoch": 3.52, "learning_rate": 4.898446833930705e-05, "loss": 0.7075, "step": 110 }, { "epoch": 3.84, "learning_rate": 4.8387096774193554e-05, "loss": 0.6554, "step": 120 }, { "epoch": 4.0, "eval_accuracy": 0.7886133032694476, "eval_f1": 0.7773123934332622, "eval_loss": 0.633278489112854, "eval_precision": 0.8070560215205211, "eval_recall": 0.7886133032694476, "eval_runtime": 10.9718, "eval_samples_per_second": 161.688, "eval_steps_per_second": 1.276, "step": 125 }, { "epoch": 4.16, "learning_rate": 4.778972520908005e-05, "loss": 0.6531, "step": 130 }, { "epoch": 4.48, "learning_rate": 4.7192353643966546e-05, "loss": 0.6259, "step": 140 }, { "epoch": 4.8, "learning_rate": 4.659498207885305e-05, "loss": 0.6342, "step": 150 }, { "epoch": 4.99, "eval_accuracy": 0.8134160090191658, "eval_f1": 0.8137533775668285, "eval_loss": 0.5532479882240295, "eval_precision": 0.8197652262902653, "eval_recall": 0.8134160090191658, "eval_runtime": 10.9592, "eval_samples_per_second": 161.874, "eval_steps_per_second": 1.277, "step": 156 }, { "epoch": 5.12, "learning_rate": 4.5997610513739546e-05, "loss": 0.5988, "step": 160 }, { "epoch": 5.44, "learning_rate": 4.540023894862604e-05, "loss": 0.6068, "step": 170 }, { "epoch": 5.76, "learning_rate": 4.4802867383512545e-05, "loss": 0.5565, "step": 180 }, { "epoch": 5.98, "eval_accuracy": 0.8207440811724915, "eval_f1": 0.8179640665669187, "eval_loss": 0.5323979258537292, "eval_precision": 0.8343261000652813, "eval_recall": 0.8207440811724915, "eval_runtime": 11.3197, "eval_samples_per_second": 156.718, "eval_steps_per_second": 1.237, "step": 187 }, { "epoch": 6.08, "learning_rate": 4.420549581839905e-05, "loss": 0.5592, "step": 190 }, { "epoch": 6.4, "learning_rate": 4.360812425328555e-05, "loss": 0.5665, "step": 200 }, { "epoch": 6.72, "learning_rate": 4.301075268817205e-05, "loss": 0.5475, "step": 210 }, { "epoch": 6.98, "eval_accuracy": 0.8354002254791432, "eval_f1": 0.8343016406233452, "eval_loss": 0.5029850602149963, "eval_precision": 0.8387498718572008, "eval_recall": 0.8354002254791432, "eval_runtime": 11.3637, "eval_samples_per_second": 156.112, "eval_steps_per_second": 1.232, "step": 218 }, { "epoch": 7.04, "learning_rate": 4.241338112305854e-05, "loss": 0.5693, "step": 220 }, { "epoch": 7.36, "learning_rate": 4.1816009557945046e-05, "loss": 0.5376, "step": 230 }, { "epoch": 7.68, "learning_rate": 4.121863799283154e-05, "loss": 0.5441, "step": 240 }, { "epoch": 8.0, "learning_rate": 4.062126642771804e-05, "loss": 0.5271, "step": 250 }, { "epoch": 8.0, "eval_accuracy": 0.8337091319052987, "eval_f1": 0.832521887921682, "eval_loss": 0.4833582043647766, "eval_precision": 0.8443837910122144, "eval_recall": 0.8337091319052987, "eval_runtime": 10.9142, "eval_samples_per_second": 162.54, "eval_steps_per_second": 1.283, "step": 250 }, { "epoch": 8.32, "learning_rate": 4.002389486260454e-05, "loss": 0.4638, "step": 260 }, { "epoch": 8.64, "learning_rate": 3.9426523297491045e-05, "loss": 0.5195, "step": 270 }, { "epoch": 8.96, "learning_rate": 3.882915173237754e-05, "loss": 0.5086, "step": 280 }, { "epoch": 8.99, "eval_accuracy": 0.8432919954904171, "eval_f1": 0.838178160275144, "eval_loss": 0.4611281752586365, "eval_precision": 0.8440756169112026, "eval_recall": 0.8432919954904171, "eval_runtime": 10.9989, "eval_samples_per_second": 161.29, "eval_steps_per_second": 1.273, "step": 281 }, { "epoch": 9.28, "learning_rate": 3.8231780167264044e-05, "loss": 0.4719, "step": 290 }, { "epoch": 9.6, "learning_rate": 3.763440860215054e-05, "loss": 0.5024, "step": 300 }, { "epoch": 9.92, "learning_rate": 3.7037037037037037e-05, "loss": 0.4341, "step": 310 }, { "epoch": 9.98, "eval_accuracy": 0.850620067643743, "eval_f1": 0.8508828796406267, "eval_loss": 0.44368883967399597, "eval_precision": 0.853874747203978, "eval_recall": 0.850620067643743, "eval_runtime": 10.9558, "eval_samples_per_second": 161.923, "eval_steps_per_second": 1.278, "step": 312 }, { "epoch": 10.24, "learning_rate": 3.643966547192354e-05, "loss": 0.4465, "step": 320 }, { "epoch": 10.56, "learning_rate": 3.5842293906810036e-05, "loss": 0.4517, "step": 330 }, { "epoch": 10.88, "learning_rate": 3.524492234169653e-05, "loss": 0.4557, "step": 340 }, { "epoch": 10.98, "eval_accuracy": 0.8483652762119503, "eval_f1": 0.8495285854672462, "eval_loss": 0.45592719316482544, "eval_precision": 0.8528985684910855, "eval_recall": 0.8483652762119503, "eval_runtime": 11.0155, "eval_samples_per_second": 161.046, "eval_steps_per_second": 1.271, "step": 343 }, { "epoch": 11.2, "learning_rate": 3.4647550776583035e-05, "loss": 0.4113, "step": 350 }, { "epoch": 11.52, "learning_rate": 3.405017921146954e-05, "loss": 0.4091, "step": 360 }, { "epoch": 11.84, "learning_rate": 3.3452807646356034e-05, "loss": 0.4179, "step": 370 }, { "epoch": 12.0, "eval_accuracy": 0.8128523111612176, "eval_f1": 0.8133198423401731, "eval_loss": 0.5942395329475403, "eval_precision": 0.8256669174022289, "eval_recall": 0.8128523111612176, "eval_runtime": 11.1803, "eval_samples_per_second": 158.672, "eval_steps_per_second": 1.252, "step": 375 }, { "epoch": 12.16, "learning_rate": 3.285543608124254e-05, "loss": 0.4124, "step": 380 }, { "epoch": 12.48, "learning_rate": 3.2258064516129034e-05, "loss": 0.3914, "step": 390 }, { "epoch": 12.8, "learning_rate": 3.1660692951015537e-05, "loss": 0.4243, "step": 400 }, { "epoch": 12.99, "eval_accuracy": 0.8540022547914318, "eval_f1": 0.8518108652158739, "eval_loss": 0.459916353225708, "eval_precision": 0.8536974522219319, "eval_recall": 0.8540022547914318, "eval_runtime": 10.9326, "eval_samples_per_second": 162.267, "eval_steps_per_second": 1.281, "step": 406 }, { "epoch": 13.12, "learning_rate": 3.106332138590203e-05, "loss": 0.382, "step": 410 }, { "epoch": 13.44, "learning_rate": 3.046594982078853e-05, "loss": 0.385, "step": 420 }, { "epoch": 13.76, "learning_rate": 2.9868578255675032e-05, "loss": 0.372, "step": 430 }, { "epoch": 13.98, "eval_accuracy": 0.8410372040586246, "eval_f1": 0.8402829432875131, "eval_loss": 0.4743349850177765, "eval_precision": 0.8471728150244378, "eval_recall": 0.8410372040586246, "eval_runtime": 10.9817, "eval_samples_per_second": 161.541, "eval_steps_per_second": 1.275, "step": 437 }, { "epoch": 14.08, "learning_rate": 2.9271206690561532e-05, "loss": 0.4001, "step": 440 }, { "epoch": 14.4, "learning_rate": 2.8673835125448028e-05, "loss": 0.3564, "step": 450 }, { "epoch": 14.72, "learning_rate": 2.807646356033453e-05, "loss": 0.4003, "step": 460 }, { "epoch": 14.98, "eval_accuracy": 0.8478015783540023, "eval_f1": 0.8460587531419033, "eval_loss": 0.47490546107292175, "eval_precision": 0.8470790221196264, "eval_recall": 0.8478015783540023, "eval_runtime": 10.9186, "eval_samples_per_second": 162.475, "eval_steps_per_second": 1.282, "step": 468 }, { "epoch": 15.04, "learning_rate": 2.747909199522103e-05, "loss": 0.3709, "step": 470 }, { "epoch": 15.36, "learning_rate": 2.6881720430107527e-05, "loss": 0.3343, "step": 480 }, { "epoch": 15.68, "learning_rate": 2.628434886499403e-05, "loss": 0.342, "step": 490 }, { "epoch": 16.0, "learning_rate": 2.5686977299880526e-05, "loss": 0.344, "step": 500 }, { "epoch": 16.0, "eval_accuracy": 0.8596392333709132, "eval_f1": 0.8572223333441049, "eval_loss": 0.4677709937095642, "eval_precision": 0.8574700407213602, "eval_recall": 0.8596392333709132, "eval_runtime": 11.0483, "eval_samples_per_second": 160.568, "eval_steps_per_second": 1.267, "step": 500 }, { "epoch": 16.32, "learning_rate": 2.5089605734767026e-05, "loss": 0.3188, "step": 510 }, { "epoch": 16.64, "learning_rate": 2.4492234169653525e-05, "loss": 0.3423, "step": 520 }, { "epoch": 16.96, "learning_rate": 2.3894862604540025e-05, "loss": 0.3252, "step": 530 }, { "epoch": 16.99, "eval_accuracy": 0.8472378804960541, "eval_f1": 0.8458902825091352, "eval_loss": 0.5024080276489258, "eval_precision": 0.8469505736783929, "eval_recall": 0.8472378804960541, "eval_runtime": 10.8627, "eval_samples_per_second": 163.312, "eval_steps_per_second": 1.289, "step": 531 }, { "epoch": 17.28, "learning_rate": 2.3297491039426525e-05, "loss": 0.3194, "step": 540 }, { "epoch": 17.6, "learning_rate": 2.270011947431302e-05, "loss": 0.315, "step": 550 }, { "epoch": 17.92, "learning_rate": 2.2102747909199524e-05, "loss": 0.3166, "step": 560 }, { "epoch": 17.98, "eval_accuracy": 0.8438556933483653, "eval_f1": 0.841753361423925, "eval_loss": 0.5038200616836548, "eval_precision": 0.8441624506359545, "eval_recall": 0.8438556933483653, "eval_runtime": 10.9467, "eval_samples_per_second": 162.059, "eval_steps_per_second": 1.279, "step": 562 }, { "epoch": 18.24, "learning_rate": 2.1505376344086024e-05, "loss": 0.3036, "step": 570 }, { "epoch": 18.56, "learning_rate": 2.0908004778972523e-05, "loss": 0.2912, "step": 580 }, { "epoch": 18.88, "learning_rate": 2.031063321385902e-05, "loss": 0.2978, "step": 590 }, { "epoch": 18.98, "eval_accuracy": 0.8365276211950394, "eval_f1": 0.8349266239501303, "eval_loss": 0.5239560008049011, "eval_precision": 0.8350616045263993, "eval_recall": 0.8365276211950394, "eval_runtime": 10.8783, "eval_samples_per_second": 163.078, "eval_steps_per_second": 1.287, "step": 593 }, { "epoch": 19.2, "learning_rate": 1.9713261648745522e-05, "loss": 0.2865, "step": 600 }, { "epoch": 19.52, "learning_rate": 1.9115890083632022e-05, "loss": 0.2903, "step": 610 }, { "epoch": 19.84, "learning_rate": 1.8518518518518518e-05, "loss": 0.2748, "step": 620 }, { "epoch": 20.0, "eval_accuracy": 0.8511837655016911, "eval_f1": 0.8472172420526128, "eval_loss": 0.5176008343696594, "eval_precision": 0.849662483048973, "eval_recall": 0.8511837655016911, "eval_runtime": 10.9331, "eval_samples_per_second": 162.259, "eval_steps_per_second": 1.281, "step": 625 }, { "epoch": 20.16, "learning_rate": 1.7921146953405018e-05, "loss": 0.2696, "step": 630 }, { "epoch": 20.48, "learning_rate": 1.7323775388291518e-05, "loss": 0.2645, "step": 640 }, { "epoch": 20.8, "learning_rate": 1.6726403823178017e-05, "loss": 0.2691, "step": 650 }, { "epoch": 20.99, "eval_accuracy": 0.8534385569334837, "eval_f1": 0.850574302117733, "eval_loss": 0.5529418587684631, "eval_precision": 0.8514327131139023, "eval_recall": 0.8534385569334837, "eval_runtime": 11.1461, "eval_samples_per_second": 159.159, "eval_steps_per_second": 1.256, "step": 656 }, { "epoch": 21.12, "learning_rate": 1.6129032258064517e-05, "loss": 0.2555, "step": 660 }, { "epoch": 21.44, "learning_rate": 1.5531660692951016e-05, "loss": 0.2532, "step": 670 }, { "epoch": 21.76, "learning_rate": 1.4934289127837516e-05, "loss": 0.2571, "step": 680 }, { "epoch": 21.98, "eval_accuracy": 0.8562570462232244, "eval_f1": 0.8535331719575036, "eval_loss": 0.5440572500228882, "eval_precision": 0.8573345104741896, "eval_recall": 0.8562570462232244, "eval_runtime": 10.8704, "eval_samples_per_second": 163.196, "eval_steps_per_second": 1.288, "step": 687 }, { "epoch": 22.08, "learning_rate": 1.4336917562724014e-05, "loss": 0.267, "step": 690 }, { "epoch": 22.4, "learning_rate": 1.3739545997610515e-05, "loss": 0.2577, "step": 700 }, { "epoch": 22.72, "learning_rate": 1.3142174432497015e-05, "loss": 0.2451, "step": 710 }, { "epoch": 22.98, "eval_accuracy": 0.842728297632469, "eval_f1": 0.8411701610062584, "eval_loss": 0.5440482497215271, "eval_precision": 0.8426930970925739, "eval_recall": 0.842728297632469, "eval_runtime": 11.1949, "eval_samples_per_second": 158.466, "eval_steps_per_second": 1.251, "step": 718 }, { "epoch": 23.04, "learning_rate": 1.2544802867383513e-05, "loss": 0.2448, "step": 720 }, { "epoch": 23.36, "learning_rate": 1.1947431302270013e-05, "loss": 0.2316, "step": 730 }, { "epoch": 23.68, "learning_rate": 1.135005973715651e-05, "loss": 0.2545, "step": 740 }, { "epoch": 24.0, "learning_rate": 1.0752688172043012e-05, "loss": 0.2256, "step": 750 }, { "epoch": 24.0, "eval_accuracy": 0.850620067643743, "eval_f1": 0.8474199467625554, "eval_loss": 0.5489442348480225, "eval_precision": 0.8467373057809777, "eval_recall": 0.850620067643743, "eval_runtime": 11.283, "eval_samples_per_second": 157.228, "eval_steps_per_second": 1.241, "step": 750 }, { "epoch": 24.32, "learning_rate": 1.015531660692951e-05, "loss": 0.2449, "step": 760 }, { "epoch": 24.64, "learning_rate": 9.557945041816011e-06, "loss": 0.2227, "step": 770 }, { "epoch": 24.96, "learning_rate": 8.960573476702509e-06, "loss": 0.2304, "step": 780 }, { "epoch": 24.99, "eval_accuracy": 0.8534385569334837, "eval_f1": 0.8498128597008927, "eval_loss": 0.5695037245750427, "eval_precision": 0.8492335232117652, "eval_recall": 0.8534385569334837, "eval_runtime": 10.9266, "eval_samples_per_second": 162.356, "eval_steps_per_second": 1.281, "step": 781 }, { "epoch": 25.28, "learning_rate": 8.363201911589009e-06, "loss": 0.2085, "step": 790 }, { "epoch": 25.6, "learning_rate": 7.765830346475508e-06, "loss": 0.2208, "step": 800 }, { "epoch": 25.92, "learning_rate": 7.168458781362007e-06, "loss": 0.2102, "step": 810 }, { "epoch": 25.98, "eval_accuracy": 0.8568207440811725, "eval_f1": 0.8540004867753292, "eval_loss": 0.5347083210945129, "eval_precision": 0.8532798699553742, "eval_recall": 0.8568207440811725, "eval_runtime": 10.9611, "eval_samples_per_second": 161.845, "eval_steps_per_second": 1.277, "step": 812 }, { "epoch": 26.24, "learning_rate": 6.5710872162485075e-06, "loss": 0.2199, "step": 820 }, { "epoch": 26.56, "learning_rate": 5.973715651135006e-06, "loss": 0.2082, "step": 830 }, { "epoch": 26.88, "learning_rate": 5.376344086021506e-06, "loss": 0.2172, "step": 840 }, { "epoch": 26.98, "eval_accuracy": 0.8523111612175873, "eval_f1": 0.8517977660010669, "eval_loss": 0.5399231910705566, "eval_precision": 0.8525415835506752, "eval_recall": 0.8523111612175873, "eval_runtime": 11.1925, "eval_samples_per_second": 158.5, "eval_steps_per_second": 1.251, "step": 843 }, { "epoch": 27.2, "learning_rate": 4.7789725209080055e-06, "loss": 0.2079, "step": 850 }, { "epoch": 27.52, "learning_rate": 4.181600955794504e-06, "loss": 0.2186, "step": 860 }, { "epoch": 27.84, "learning_rate": 3.5842293906810035e-06, "loss": 0.1953, "step": 870 }, { "epoch": 28.0, "eval_accuracy": 0.855129650507328, "eval_f1": 0.8521518627937925, "eval_loss": 0.5699470043182373, "eval_precision": 0.8547328115878524, "eval_recall": 0.855129650507328, "eval_runtime": 11.0392, "eval_samples_per_second": 160.7, "eval_steps_per_second": 1.268, "step": 875 }, { "epoch": 28.16, "learning_rate": 2.986857825567503e-06, "loss": 0.2106, "step": 880 }, { "epoch": 28.48, "learning_rate": 2.3894862604540028e-06, "loss": 0.1925, "step": 890 }, { "epoch": 28.8, "learning_rate": 1.7921146953405017e-06, "loss": 0.2035, "step": 900 }, { "epoch": 28.99, "eval_accuracy": 0.863021420518602, "eval_f1": 0.8606635079855887, "eval_loss": 0.5538451075553894, "eval_precision": 0.863028156731382, "eval_recall": 0.863021420518602, "eval_runtime": 11.164, "eval_samples_per_second": 158.904, "eval_steps_per_second": 1.254, "step": 906 }, { "epoch": 29.12, "learning_rate": 1.1947431302270014e-06, "loss": 0.1954, "step": 910 }, { "epoch": 29.44, "learning_rate": 5.973715651135007e-07, "loss": 0.186, "step": 920 }, { "epoch": 29.76, "learning_rate": 0.0, "loss": 0.1926, "step": 930 }, { "epoch": 29.76, "eval_accuracy": 0.863021420518602, "eval_f1": 0.860579135743333, "eval_loss": 0.543451189994812, "eval_precision": 0.8608767769841421, "eval_recall": 0.863021420518602, "eval_runtime": 11.0366, "eval_samples_per_second": 160.739, "eval_steps_per_second": 1.269, "step": 930 }, { "epoch": 29.76, "step": 930, "total_flos": 4.852841995091313e+19, "train_loss": 0.44145851263435937, "train_runtime": 6147.3767, "train_samples_per_second": 77.897, "train_steps_per_second": 0.151 } ], "logging_steps": 10, "max_steps": 930, "num_input_tokens_seen": 0, "num_train_epochs": 30, "save_steps": 500, "total_flos": 4.852841995091313e+19, "train_batch_size": 128, "trial_name": null, "trial_params": null }