{ "best_metric": 0.11452386528253555, "best_model_checkpoint": "./vitforimcls/checkpoint-1500", "epoch": 4.0, "eval_steps": 100, "global_step": 1500, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.03, "learning_rate": 0.00019866666666666668, "loss": 1.975, "step": 10 }, { "epoch": 0.05, "learning_rate": 0.00019733333333333335, "loss": 1.1804, "step": 20 }, { "epoch": 0.08, "learning_rate": 0.000196, "loss": 0.7784, "step": 30 }, { "epoch": 0.11, "learning_rate": 0.0001946666666666667, "loss": 0.526, "step": 40 }, { "epoch": 0.13, "learning_rate": 0.00019333333333333333, "loss": 0.4219, "step": 50 }, { "epoch": 0.16, "learning_rate": 0.000192, "loss": 0.3939, "step": 60 }, { "epoch": 0.19, "learning_rate": 0.00019066666666666668, "loss": 0.4119, "step": 70 }, { "epoch": 0.21, "learning_rate": 0.00018933333333333335, "loss": 0.3594, "step": 80 }, { "epoch": 0.24, "learning_rate": 0.000188, "loss": 0.3315, "step": 90 }, { "epoch": 0.27, "learning_rate": 0.0001866666666666667, "loss": 0.2995, "step": 100 }, { "epoch": 0.27, "eval_accuracy": 0.9108333333333334, "eval_loss": 0.3419191539287567, "eval_runtime": 221.995, "eval_samples_per_second": 27.028, "eval_steps_per_second": 0.423, "step": 100 }, { "epoch": 0.29, "learning_rate": 0.00018533333333333333, "loss": 0.2941, "step": 110 }, { "epoch": 0.32, "learning_rate": 0.00018400000000000003, "loss": 0.294, "step": 120 }, { "epoch": 0.35, "learning_rate": 0.00018266666666666667, "loss": 0.2903, "step": 130 }, { "epoch": 0.37, "learning_rate": 0.00018133333333333334, "loss": 0.2592, "step": 140 }, { "epoch": 0.4, "learning_rate": 0.00018, "loss": 0.2496, "step": 150 }, { "epoch": 0.43, "learning_rate": 0.00017866666666666668, "loss": 0.265, "step": 160 }, { "epoch": 0.45, "learning_rate": 0.00017733333333333335, "loss": 0.245, "step": 170 }, { "epoch": 0.48, "learning_rate": 0.00017600000000000002, "loss": 0.2321, "step": 180 }, { "epoch": 0.51, "learning_rate": 0.00017466666666666667, "loss": 0.2008, "step": 190 }, { "epoch": 0.53, "learning_rate": 0.00017333333333333334, "loss": 0.2289, "step": 200 }, { "epoch": 0.53, "eval_accuracy": 0.9288333333333333, "eval_loss": 0.24824613332748413, "eval_runtime": 222.622, "eval_samples_per_second": 26.952, "eval_steps_per_second": 0.422, "step": 200 }, { "epoch": 0.56, "learning_rate": 0.000172, "loss": 0.2231, "step": 210 }, { "epoch": 0.59, "learning_rate": 0.00017066666666666668, "loss": 0.2198, "step": 220 }, { "epoch": 0.61, "learning_rate": 0.00016933333333333335, "loss": 0.2189, "step": 230 }, { "epoch": 0.64, "learning_rate": 0.000168, "loss": 0.2681, "step": 240 }, { "epoch": 0.67, "learning_rate": 0.0001666666666666667, "loss": 0.1843, "step": 250 }, { "epoch": 0.69, "learning_rate": 0.00016533333333333333, "loss": 0.219, "step": 260 }, { "epoch": 0.72, "learning_rate": 0.000164, "loss": 0.1871, "step": 270 }, { "epoch": 0.75, "learning_rate": 0.00016266666666666667, "loss": 0.2011, "step": 280 }, { "epoch": 0.77, "learning_rate": 0.00016133333333333334, "loss": 0.2038, "step": 290 }, { "epoch": 0.8, "learning_rate": 0.00016, "loss": 0.1811, "step": 300 }, { "epoch": 0.8, "eval_accuracy": 0.9356666666666666, "eval_loss": 0.2139243632555008, "eval_runtime": 223.6566, "eval_samples_per_second": 26.827, "eval_steps_per_second": 0.42, "step": 300 }, { "epoch": 0.83, "learning_rate": 0.00015866666666666668, "loss": 0.2091, "step": 310 }, { "epoch": 0.85, "learning_rate": 0.00015733333333333333, "loss": 0.2127, "step": 320 }, { "epoch": 0.88, "learning_rate": 0.00015600000000000002, "loss": 0.1931, "step": 330 }, { "epoch": 0.91, "learning_rate": 0.00015466666666666667, "loss": 0.1942, "step": 340 }, { "epoch": 0.93, "learning_rate": 0.00015333333333333334, "loss": 0.1704, "step": 350 }, { "epoch": 0.96, "learning_rate": 0.000152, "loss": 0.1733, "step": 360 }, { "epoch": 0.99, "learning_rate": 0.00015066666666666668, "loss": 0.1759, "step": 370 }, { "epoch": 1.01, "learning_rate": 0.00014933333333333335, "loss": 0.1403, "step": 380 }, { "epoch": 1.04, "learning_rate": 0.000148, "loss": 0.0968, "step": 390 }, { "epoch": 1.07, "learning_rate": 0.00014666666666666666, "loss": 0.0797, "step": 400 }, { "epoch": 1.07, "eval_accuracy": 0.946, "eval_loss": 0.18132811784744263, "eval_runtime": 198.5359, "eval_samples_per_second": 30.221, "eval_steps_per_second": 0.473, "step": 400 }, { "epoch": 1.09, "learning_rate": 0.00014533333333333333, "loss": 0.099, "step": 410 }, { "epoch": 1.12, "learning_rate": 0.000144, "loss": 0.1239, "step": 420 }, { "epoch": 1.15, "learning_rate": 0.00014266666666666667, "loss": 0.1084, "step": 430 }, { "epoch": 1.17, "learning_rate": 0.00014133333333333334, "loss": 0.1008, "step": 440 }, { "epoch": 1.2, "learning_rate": 0.00014, "loss": 0.1114, "step": 450 }, { "epoch": 1.23, "learning_rate": 0.00013866666666666669, "loss": 0.1102, "step": 460 }, { "epoch": 1.25, "learning_rate": 0.00013733333333333333, "loss": 0.0892, "step": 470 }, { "epoch": 1.28, "learning_rate": 0.00013600000000000003, "loss": 0.081, "step": 480 }, { "epoch": 1.31, "learning_rate": 0.00013466666666666667, "loss": 0.071, "step": 490 }, { "epoch": 1.33, "learning_rate": 0.00013333333333333334, "loss": 0.1128, "step": 500 }, { "epoch": 1.33, "eval_accuracy": 0.9451666666666667, "eval_loss": 0.1741384118795395, "eval_runtime": 167.1262, "eval_samples_per_second": 35.901, "eval_steps_per_second": 0.562, "step": 500 }, { "epoch": 1.36, "learning_rate": 0.000132, "loss": 0.0966, "step": 510 }, { "epoch": 1.39, "learning_rate": 0.00013066666666666668, "loss": 0.0998, "step": 520 }, { "epoch": 1.41, "learning_rate": 0.00012933333333333332, "loss": 0.0951, "step": 530 }, { "epoch": 1.44, "learning_rate": 0.00012800000000000002, "loss": 0.0792, "step": 540 }, { "epoch": 1.47, "learning_rate": 0.00012666666666666666, "loss": 0.0928, "step": 550 }, { "epoch": 1.49, "learning_rate": 0.00012533333333333334, "loss": 0.0893, "step": 560 }, { "epoch": 1.52, "learning_rate": 0.000124, "loss": 0.103, "step": 570 }, { "epoch": 1.55, "learning_rate": 0.00012266666666666668, "loss": 0.1357, "step": 580 }, { "epoch": 1.57, "learning_rate": 0.00012133333333333335, "loss": 0.0883, "step": 590 }, { "epoch": 1.6, "learning_rate": 0.00012, "loss": 0.086, "step": 600 }, { "epoch": 1.6, "eval_accuracy": 0.9513333333333334, "eval_loss": 0.16589362919330597, "eval_runtime": 175.0929, "eval_samples_per_second": 34.268, "eval_steps_per_second": 0.537, "step": 600 }, { "epoch": 1.63, "learning_rate": 0.00011866666666666669, "loss": 0.091, "step": 610 }, { "epoch": 1.65, "learning_rate": 0.00011733333333333334, "loss": 0.0701, "step": 620 }, { "epoch": 1.68, "learning_rate": 0.000116, "loss": 0.0834, "step": 630 }, { "epoch": 1.71, "learning_rate": 0.00011466666666666667, "loss": 0.0891, "step": 640 }, { "epoch": 1.73, "learning_rate": 0.00011333333333333334, "loss": 0.0896, "step": 650 }, { "epoch": 1.76, "learning_rate": 0.00011200000000000001, "loss": 0.0853, "step": 660 }, { "epoch": 1.79, "learning_rate": 0.00011066666666666667, "loss": 0.0799, "step": 670 }, { "epoch": 1.81, "learning_rate": 0.00010933333333333333, "loss": 0.0742, "step": 680 }, { "epoch": 1.84, "learning_rate": 0.00010800000000000001, "loss": 0.093, "step": 690 }, { "epoch": 1.87, "learning_rate": 0.00010666666666666667, "loss": 0.0815, "step": 700 }, { "epoch": 1.87, "eval_accuracy": 0.9546666666666667, "eval_loss": 0.14684614539146423, "eval_runtime": 174.5528, "eval_samples_per_second": 34.374, "eval_steps_per_second": 0.539, "step": 700 }, { "epoch": 1.89, "learning_rate": 0.00010533333333333332, "loss": 0.0912, "step": 710 }, { "epoch": 1.92, "learning_rate": 0.00010400000000000001, "loss": 0.0677, "step": 720 }, { "epoch": 1.95, "learning_rate": 0.00010266666666666666, "loss": 0.0806, "step": 730 }, { "epoch": 1.97, "learning_rate": 0.00010133333333333335, "loss": 0.0762, "step": 740 }, { "epoch": 2.0, "learning_rate": 0.0001, "loss": 0.0876, "step": 750 }, { "epoch": 2.03, "learning_rate": 9.866666666666668e-05, "loss": 0.0425, "step": 760 }, { "epoch": 2.05, "learning_rate": 9.733333333333335e-05, "loss": 0.054, "step": 770 }, { "epoch": 2.08, "learning_rate": 9.6e-05, "loss": 0.0384, "step": 780 }, { "epoch": 2.11, "learning_rate": 9.466666666666667e-05, "loss": 0.0506, "step": 790 }, { "epoch": 2.13, "learning_rate": 9.333333333333334e-05, "loss": 0.048, "step": 800 }, { "epoch": 2.13, "eval_accuracy": 0.9591666666666666, "eval_loss": 0.1392858624458313, "eval_runtime": 165.6771, "eval_samples_per_second": 36.215, "eval_steps_per_second": 0.567, "step": 800 }, { "epoch": 2.16, "learning_rate": 9.200000000000001e-05, "loss": 0.0346, "step": 810 }, { "epoch": 2.19, "learning_rate": 9.066666666666667e-05, "loss": 0.0258, "step": 820 }, { "epoch": 2.21, "learning_rate": 8.933333333333334e-05, "loss": 0.0385, "step": 830 }, { "epoch": 2.24, "learning_rate": 8.800000000000001e-05, "loss": 0.0239, "step": 840 }, { "epoch": 2.27, "learning_rate": 8.666666666666667e-05, "loss": 0.0327, "step": 850 }, { "epoch": 2.29, "learning_rate": 8.533333333333334e-05, "loss": 0.0334, "step": 860 }, { "epoch": 2.32, "learning_rate": 8.4e-05, "loss": 0.0318, "step": 870 }, { "epoch": 2.35, "learning_rate": 8.266666666666667e-05, "loss": 0.0307, "step": 880 }, { "epoch": 2.37, "learning_rate": 8.133333333333334e-05, "loss": 0.0267, "step": 890 }, { "epoch": 2.4, "learning_rate": 8e-05, "loss": 0.021, "step": 900 }, { "epoch": 2.4, "eval_accuracy": 0.9603333333333334, "eval_loss": 0.13989999890327454, "eval_runtime": 169.1114, "eval_samples_per_second": 35.48, "eval_steps_per_second": 0.556, "step": 900 }, { "epoch": 2.43, "learning_rate": 7.866666666666666e-05, "loss": 0.0247, "step": 910 }, { "epoch": 2.45, "learning_rate": 7.733333333333333e-05, "loss": 0.0401, "step": 920 }, { "epoch": 2.48, "learning_rate": 7.6e-05, "loss": 0.0202, "step": 930 }, { "epoch": 2.51, "learning_rate": 7.466666666666667e-05, "loss": 0.0289, "step": 940 }, { "epoch": 2.53, "learning_rate": 7.333333333333333e-05, "loss": 0.0293, "step": 950 }, { "epoch": 2.56, "learning_rate": 7.2e-05, "loss": 0.025, "step": 960 }, { "epoch": 2.59, "learning_rate": 7.066666666666667e-05, "loss": 0.016, "step": 970 }, { "epoch": 2.61, "learning_rate": 6.933333333333334e-05, "loss": 0.0315, "step": 980 }, { "epoch": 2.64, "learning_rate": 6.800000000000001e-05, "loss": 0.0309, "step": 990 }, { "epoch": 2.67, "learning_rate": 6.666666666666667e-05, "loss": 0.0271, "step": 1000 }, { "epoch": 2.67, "eval_accuracy": 0.9641666666666666, "eval_loss": 0.13339994847774506, "eval_runtime": 163.886, "eval_samples_per_second": 36.611, "eval_steps_per_second": 0.574, "step": 1000 }, { "epoch": 2.69, "learning_rate": 6.533333333333334e-05, "loss": 0.017, "step": 1010 }, { "epoch": 2.72, "learning_rate": 6.400000000000001e-05, "loss": 0.0231, "step": 1020 }, { "epoch": 2.75, "learning_rate": 6.266666666666667e-05, "loss": 0.0256, "step": 1030 }, { "epoch": 2.77, "learning_rate": 6.133333333333334e-05, "loss": 0.0166, "step": 1040 }, { "epoch": 2.8, "learning_rate": 6e-05, "loss": 0.027, "step": 1050 }, { "epoch": 2.83, "learning_rate": 5.866666666666667e-05, "loss": 0.0202, "step": 1060 }, { "epoch": 2.85, "learning_rate": 5.7333333333333336e-05, "loss": 0.0235, "step": 1070 }, { "epoch": 2.88, "learning_rate": 5.6000000000000006e-05, "loss": 0.0185, "step": 1080 }, { "epoch": 2.91, "learning_rate": 5.466666666666666e-05, "loss": 0.0167, "step": 1090 }, { "epoch": 2.93, "learning_rate": 5.333333333333333e-05, "loss": 0.0231, "step": 1100 }, { "epoch": 2.93, "eval_accuracy": 0.9658333333333333, "eval_loss": 0.12280993908643723, "eval_runtime": 172.4826, "eval_samples_per_second": 34.786, "eval_steps_per_second": 0.545, "step": 1100 }, { "epoch": 2.96, "learning_rate": 5.2000000000000004e-05, "loss": 0.031, "step": 1110 }, { "epoch": 2.99, "learning_rate": 5.0666666666666674e-05, "loss": 0.0332, "step": 1120 }, { "epoch": 3.01, "learning_rate": 4.933333333333334e-05, "loss": 0.0226, "step": 1130 }, { "epoch": 3.04, "learning_rate": 4.8e-05, "loss": 0.0094, "step": 1140 }, { "epoch": 3.07, "learning_rate": 4.666666666666667e-05, "loss": 0.0095, "step": 1150 }, { "epoch": 3.09, "learning_rate": 4.5333333333333335e-05, "loss": 0.0075, "step": 1160 }, { "epoch": 3.12, "learning_rate": 4.4000000000000006e-05, "loss": 0.0069, "step": 1170 }, { "epoch": 3.15, "learning_rate": 4.266666666666667e-05, "loss": 0.0072, "step": 1180 }, { "epoch": 3.17, "learning_rate": 4.133333333333333e-05, "loss": 0.0048, "step": 1190 }, { "epoch": 3.2, "learning_rate": 4e-05, "loss": 0.0101, "step": 1200 }, { "epoch": 3.2, "eval_accuracy": 0.9673333333333334, "eval_loss": 0.12290121614933014, "eval_runtime": 169.3314, "eval_samples_per_second": 35.433, "eval_steps_per_second": 0.555, "step": 1200 }, { "epoch": 3.23, "learning_rate": 3.866666666666667e-05, "loss": 0.0085, "step": 1210 }, { "epoch": 3.25, "learning_rate": 3.733333333333334e-05, "loss": 0.01, "step": 1220 }, { "epoch": 3.28, "learning_rate": 3.6e-05, "loss": 0.0091, "step": 1230 }, { "epoch": 3.31, "learning_rate": 3.466666666666667e-05, "loss": 0.0045, "step": 1240 }, { "epoch": 3.33, "learning_rate": 3.3333333333333335e-05, "loss": 0.0089, "step": 1250 }, { "epoch": 3.36, "learning_rate": 3.2000000000000005e-05, "loss": 0.0115, "step": 1260 }, { "epoch": 3.39, "learning_rate": 3.066666666666667e-05, "loss": 0.0051, "step": 1270 }, { "epoch": 3.41, "learning_rate": 2.9333333333333336e-05, "loss": 0.011, "step": 1280 }, { "epoch": 3.44, "learning_rate": 2.8000000000000003e-05, "loss": 0.0112, "step": 1290 }, { "epoch": 3.47, "learning_rate": 2.6666666666666667e-05, "loss": 0.0041, "step": 1300 }, { "epoch": 3.47, "eval_accuracy": 0.9675, "eval_loss": 0.11890539526939392, "eval_runtime": 173.0615, "eval_samples_per_second": 34.67, "eval_steps_per_second": 0.543, "step": 1300 }, { "epoch": 3.49, "learning_rate": 2.5333333333333337e-05, "loss": 0.0062, "step": 1310 }, { "epoch": 3.52, "learning_rate": 2.4e-05, "loss": 0.0055, "step": 1320 }, { "epoch": 3.55, "learning_rate": 2.2666666666666668e-05, "loss": 0.007, "step": 1330 }, { "epoch": 3.57, "learning_rate": 2.1333333333333335e-05, "loss": 0.0039, "step": 1340 }, { "epoch": 3.6, "learning_rate": 2e-05, "loss": 0.0062, "step": 1350 }, { "epoch": 3.63, "learning_rate": 1.866666666666667e-05, "loss": 0.0042, "step": 1360 }, { "epoch": 3.65, "learning_rate": 1.7333333333333336e-05, "loss": 0.0077, "step": 1370 }, { "epoch": 3.68, "learning_rate": 1.6000000000000003e-05, "loss": 0.0065, "step": 1380 }, { "epoch": 3.71, "learning_rate": 1.4666666666666668e-05, "loss": 0.0144, "step": 1390 }, { "epoch": 3.73, "learning_rate": 1.3333333333333333e-05, "loss": 0.0043, "step": 1400 }, { "epoch": 3.73, "eval_accuracy": 0.9683333333333334, "eval_loss": 0.11652529239654541, "eval_runtime": 173.2794, "eval_samples_per_second": 34.626, "eval_steps_per_second": 0.542, "step": 1400 }, { "epoch": 3.76, "learning_rate": 1.2e-05, "loss": 0.006, "step": 1410 }, { "epoch": 3.79, "learning_rate": 1.0666666666666667e-05, "loss": 0.0059, "step": 1420 }, { "epoch": 3.81, "learning_rate": 9.333333333333334e-06, "loss": 0.0093, "step": 1430 }, { "epoch": 3.84, "learning_rate": 8.000000000000001e-06, "loss": 0.0063, "step": 1440 }, { "epoch": 3.87, "learning_rate": 6.666666666666667e-06, "loss": 0.0044, "step": 1450 }, { "epoch": 3.89, "learning_rate": 5.333333333333334e-06, "loss": 0.0051, "step": 1460 }, { "epoch": 3.92, "learning_rate": 4.000000000000001e-06, "loss": 0.004, "step": 1470 }, { "epoch": 3.95, "learning_rate": 2.666666666666667e-06, "loss": 0.0037, "step": 1480 }, { "epoch": 3.97, "learning_rate": 1.3333333333333334e-06, "loss": 0.0039, "step": 1490 }, { "epoch": 4.0, "learning_rate": 0.0, "loss": 0.0067, "step": 1500 }, { "epoch": 4.0, "eval_accuracy": 0.9696666666666667, "eval_loss": 0.11452386528253555, "eval_runtime": 166.2179, "eval_samples_per_second": 36.097, "eval_steps_per_second": 0.566, "step": 1500 }, { "epoch": 4.0, "step": 1500, "total_flos": 1.4879528813592576e+19, "train_loss": 0.11711712082227071, "train_runtime": 8681.751, "train_samples_per_second": 22.115, "train_steps_per_second": 0.173 } ], "logging_steps": 10, "max_steps": 1500, "num_train_epochs": 4, "save_steps": 100, "total_flos": 1.4879528813592576e+19, "trial_name": null, "trial_params": null }