| { |
| "best_metric": 0.16766877472400665, |
| "best_model_checkpoint": "./vit-base-beans/checkpoint-560", |
| "epoch": 18.75, |
| "global_step": 600, |
| "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 |
| } |
| ], |
| "max_steps": 640, |
| "num_train_epochs": 20, |
| "total_flos": 1.1105739126909665e+18, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
