{ "best_metric": 0.16443897783756256, "best_model_checkpoint": "./vit-base-beans/checkpoint-640", "epoch": 20.0, "global_step": 640, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.31, "learning_rate": 9.84375e-05, "loss": 3.3252, "step": 10 }, { "epoch": 0.62, "learning_rate": 9.687500000000001e-05, "loss": 3.217, "step": 20 }, { "epoch": 0.94, "learning_rate": 9.53125e-05, "loss": 3.0799, "step": 30 }, { "epoch": 1.25, "learning_rate": 9.375e-05, "loss": 2.8394, "step": 40 }, { "epoch": 1.25, "eval_accuracy": 0.4815668202764977, "eval_loss": 2.7347912788391113, "eval_runtime": 5.6305, "eval_samples_per_second": 77.081, "eval_steps_per_second": 9.768, "step": 40 }, { "epoch": 1.56, "learning_rate": 9.21875e-05, "loss": 2.598, "step": 50 }, { "epoch": 1.88, "learning_rate": 9.062500000000001e-05, "loss": 2.417, "step": 60 }, { "epoch": 2.19, "learning_rate": 8.90625e-05, "loss": 2.111, "step": 70 }, { "epoch": 2.5, "learning_rate": 8.75e-05, "loss": 1.9174, "step": 80 }, { "epoch": 2.5, "eval_accuracy": 0.8963133640552995, "eval_loss": 1.8127654790878296, "eval_runtime": 5.3595, "eval_samples_per_second": 80.978, "eval_steps_per_second": 10.262, "step": 80 }, { "epoch": 2.81, "learning_rate": 8.593750000000001e-05, "loss": 1.666, "step": 90 }, { "epoch": 3.12, "learning_rate": 8.4375e-05, "loss": 1.5159, "step": 100 }, { "epoch": 3.44, "learning_rate": 8.28125e-05, "loss": 1.3105, "step": 110 }, { "epoch": 3.75, "learning_rate": 8.125000000000001e-05, "loss": 1.1859, "step": 120 }, { "epoch": 3.75, "eval_accuracy": 0.9470046082949308, "eval_loss": 1.1414676904678345, "eval_runtime": 5.3582, "eval_samples_per_second": 80.997, "eval_steps_per_second": 10.265, "step": 120 }, { "epoch": 4.06, "learning_rate": 7.96875e-05, "loss": 1.0525, "step": 130 }, { "epoch": 4.38, "learning_rate": 7.8125e-05, "loss": 0.9085, "step": 140 }, { "epoch": 4.69, "learning_rate": 7.65625e-05, "loss": 0.8207, "step": 150 }, { "epoch": 5.0, "learning_rate": 7.500000000000001e-05, "loss": 0.7413, "step": 160 }, { "epoch": 5.0, "eval_accuracy": 0.9654377880184332, "eval_loss": 0.7720305919647217, "eval_runtime": 5.6625, "eval_samples_per_second": 76.645, "eval_steps_per_second": 9.713, "step": 160 }, { "epoch": 5.31, "learning_rate": 7.34375e-05, "loss": 0.6334, "step": 170 }, { "epoch": 5.62, "learning_rate": 7.1875e-05, "loss": 0.6058, "step": 180 }, { "epoch": 5.94, "learning_rate": 7.031250000000001e-05, "loss": 0.5319, "step": 190 }, { "epoch": 6.25, "learning_rate": 6.875e-05, "loss": 0.4761, "step": 200 }, { "epoch": 6.25, "eval_accuracy": 0.9838709677419355, "eval_loss": 0.5084273815155029, "eval_runtime": 5.4846, "eval_samples_per_second": 79.131, "eval_steps_per_second": 10.028, "step": 200 }, { "epoch": 6.56, "learning_rate": 6.71875e-05, "loss": 0.4109, "step": 210 }, { "epoch": 6.88, "learning_rate": 6.562500000000001e-05, "loss": 0.3707, "step": 220 }, { "epoch": 7.19, "learning_rate": 6.40625e-05, "loss": 0.3536, "step": 230 }, { "epoch": 7.5, "learning_rate": 6.25e-05, "loss": 0.3108, "step": 240 }, { "epoch": 7.5, "eval_accuracy": 0.9746543778801844, "eval_loss": 0.36055463552474976, "eval_runtime": 5.6696, "eval_samples_per_second": 76.549, "eval_steps_per_second": 9.701, "step": 240 }, { "epoch": 7.81, "learning_rate": 6.0937500000000004e-05, "loss": 0.2789, "step": 250 }, { "epoch": 8.12, "learning_rate": 5.9375e-05, "loss": 0.2905, "step": 260 }, { "epoch": 8.44, "learning_rate": 5.78125e-05, "loss": 0.2462, "step": 270 }, { "epoch": 8.75, "learning_rate": 5.6250000000000005e-05, "loss": 0.251, "step": 280 }, { "epoch": 8.75, "eval_accuracy": 0.9769585253456221, "eval_loss": 0.2958492040634155, "eval_runtime": 7.6402, "eval_samples_per_second": 56.805, "eval_steps_per_second": 7.199, "step": 280 }, { "epoch": 9.06, "learning_rate": 5.46875e-05, "loss": 0.2171, "step": 290 }, { "epoch": 9.38, "learning_rate": 5.3125000000000004e-05, "loss": 0.2064, "step": 300 }, { "epoch": 9.69, "learning_rate": 5.15625e-05, "loss": 0.2116, "step": 310 }, { "epoch": 10.0, "learning_rate": 5e-05, "loss": 0.1896, "step": 320 }, { "epoch": 10.0, "eval_accuracy": 0.9769585253456221, "eval_loss": 0.24788345396518707, "eval_runtime": 5.2479, "eval_samples_per_second": 82.699, "eval_steps_per_second": 10.48, "step": 320 }, { "epoch": 10.31, "learning_rate": 4.8437500000000005e-05, "loss": 0.1783, "step": 330 }, { "epoch": 10.62, "learning_rate": 4.6875e-05, "loss": 0.1859, "step": 340 }, { "epoch": 10.94, "learning_rate": 4.5312500000000004e-05, "loss": 0.1705, "step": 350 }, { "epoch": 11.25, "learning_rate": 4.375e-05, "loss": 0.1659, "step": 360 }, { "epoch": 11.25, "eval_accuracy": 0.9838709677419355, "eval_loss": 0.23752211034297943, "eval_runtime": 5.7417, "eval_samples_per_second": 75.588, "eval_steps_per_second": 9.579, "step": 360 }, { "epoch": 11.56, "learning_rate": 4.21875e-05, "loss": 0.1753, "step": 370 }, { "epoch": 11.88, "learning_rate": 4.0625000000000005e-05, "loss": 0.1509, "step": 380 }, { "epoch": 12.19, "learning_rate": 3.90625e-05, "loss": 0.1447, "step": 390 }, { "epoch": 12.5, "learning_rate": 3.7500000000000003e-05, "loss": 0.1401, "step": 400 }, { "epoch": 12.5, "eval_accuracy": 0.9792626728110599, "eval_loss": 0.20333679020404816, "eval_runtime": 5.2814, "eval_samples_per_second": 82.175, "eval_steps_per_second": 10.414, "step": 400 }, { "epoch": 12.81, "learning_rate": 3.59375e-05, "loss": 0.1508, "step": 410 }, { "epoch": 13.12, "learning_rate": 3.4375e-05, "loss": 0.1364, "step": 420 }, { "epoch": 13.44, "learning_rate": 3.2812500000000005e-05, "loss": 0.1415, "step": 430 }, { "epoch": 13.75, "learning_rate": 3.125e-05, "loss": 0.131, "step": 440 }, { "epoch": 13.75, "eval_accuracy": 0.9792626728110599, "eval_loss": 0.19693893194198608, "eval_runtime": 5.0279, "eval_samples_per_second": 86.318, "eval_steps_per_second": 10.939, "step": 440 }, { "epoch": 14.06, "learning_rate": 2.96875e-05, "loss": 0.1264, "step": 450 }, { "epoch": 14.38, "learning_rate": 2.8125000000000003e-05, "loss": 0.1376, "step": 460 }, { "epoch": 14.69, "learning_rate": 2.6562500000000002e-05, "loss": 0.1211, "step": 470 }, { "epoch": 15.0, "learning_rate": 2.5e-05, "loss": 0.1162, "step": 480 }, { "epoch": 15.0, "eval_accuracy": 0.9792626728110599, "eval_loss": 0.1791529506444931, "eval_runtime": 4.2132, "eval_samples_per_second": 103.009, "eval_steps_per_second": 13.054, "step": 480 }, { "epoch": 15.31, "learning_rate": 2.34375e-05, "loss": 0.1285, "step": 490 }, { "epoch": 15.62, "learning_rate": 2.1875e-05, "loss": 0.1136, "step": 500 }, { "epoch": 15.94, "learning_rate": 2.0312500000000002e-05, "loss": 0.1117, "step": 510 }, { "epoch": 16.25, "learning_rate": 1.8750000000000002e-05, "loss": 0.11, "step": 520 }, { "epoch": 16.25, "eval_accuracy": 0.9792626728110599, "eval_loss": 0.17193575203418732, "eval_runtime": 4.2988, "eval_samples_per_second": 100.957, "eval_steps_per_second": 12.794, "step": 520 }, { "epoch": 16.56, "learning_rate": 1.71875e-05, "loss": 0.1087, "step": 530 }, { "epoch": 16.88, "learning_rate": 1.5625e-05, "loss": 0.1065, "step": 540 }, { "epoch": 17.19, "learning_rate": 1.4062500000000001e-05, "loss": 0.1207, "step": 550 }, { "epoch": 17.5, "learning_rate": 1.25e-05, "loss": 0.1056, "step": 560 }, { "epoch": 17.5, "eval_accuracy": 0.9792626728110599, "eval_loss": 0.16766877472400665, "eval_runtime": 5.7239, "eval_samples_per_second": 75.823, "eval_steps_per_second": 9.609, "step": 560 }, { "epoch": 17.81, "learning_rate": 1.09375e-05, "loss": 0.1051, "step": 570 }, { "epoch": 18.12, "learning_rate": 9.375000000000001e-06, "loss": 0.1148, "step": 580 }, { "epoch": 18.44, "learning_rate": 7.8125e-06, "loss": 0.1043, "step": 590 }, { "epoch": 18.75, "learning_rate": 6.25e-06, "loss": 0.1128, "step": 600 }, { "epoch": 18.75, "eval_accuracy": 0.9723502304147466, "eval_loss": 0.17274537682533264, "eval_runtime": 5.5073, "eval_samples_per_second": 78.805, "eval_steps_per_second": 9.987, "step": 600 }, { "epoch": 19.06, "learning_rate": 4.6875000000000004e-06, "loss": 0.102, "step": 610 }, { "epoch": 19.38, "learning_rate": 3.125e-06, "loss": 0.1028, "step": 620 }, { "epoch": 19.69, "learning_rate": 1.5625e-06, "loss": 0.1096, "step": 630 }, { "epoch": 20.0, "learning_rate": 0.0, "loss": 0.1018, "step": 640 }, { "epoch": 20.0, "eval_accuracy": 0.9792626728110599, "eval_loss": 0.16443897783756256, "eval_runtime": 4.9159, "eval_samples_per_second": 88.285, "eval_steps_per_second": 11.188, "step": 640 }, { "epoch": 20.0, "step": 640, "total_flos": 1.1843641391623373e+18, "train_loss": 0.6315841894596815, "train_runtime": 473.5366, "train_samples_per_second": 32.268, "train_steps_per_second": 1.352 } ], "max_steps": 640, "num_train_epochs": 20, "total_flos": 1.1843641391623373e+18, "trial_name": null, "trial_params": null }