| { |
| "best_metric": 0.996, |
| "best_model_checkpoint": "AI_ImageClassification_SDXL/checkpoint-586", |
| "epoch": 0.9995735607675906, |
| "eval_steps": 500, |
| "global_step": 586, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.02, |
| "learning_rate": 2.840909090909091e-06, |
| "loss": 1.2947, |
| "step": 10 |
| }, |
| { |
| "epoch": 0.03, |
| "learning_rate": 5.681818181818182e-06, |
| "loss": 0.689, |
| "step": 20 |
| }, |
| { |
| "epoch": 0.05, |
| "learning_rate": 8.522727272727273e-06, |
| "loss": 0.2576, |
| "step": 30 |
| }, |
| { |
| "epoch": 0.07, |
| "learning_rate": 1.1363636363636365e-05, |
| "loss": 0.23, |
| "step": 40 |
| }, |
| { |
| "epoch": 0.09, |
| "learning_rate": 1.4204545454545456e-05, |
| "loss": 0.1137, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.1, |
| "learning_rate": 1.7045454545454546e-05, |
| "loss": 0.0554, |
| "step": 60 |
| }, |
| { |
| "epoch": 0.12, |
| "learning_rate": 1.9886363636363638e-05, |
| "loss": 0.059, |
| "step": 70 |
| }, |
| { |
| "epoch": 0.14, |
| "learning_rate": 2.272727272727273e-05, |
| "loss": 0.0583, |
| "step": 80 |
| }, |
| { |
| "epoch": 0.15, |
| "learning_rate": 2.5568181818181817e-05, |
| "loss": 0.0349, |
| "step": 90 |
| }, |
| { |
| "epoch": 0.17, |
| "learning_rate": 2.8409090909090912e-05, |
| "loss": 0.0864, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.19, |
| "learning_rate": 3.125e-05, |
| "loss": 0.0145, |
| "step": 110 |
| }, |
| { |
| "epoch": 0.2, |
| "learning_rate": 3.409090909090909e-05, |
| "loss": 0.0358, |
| "step": 120 |
| }, |
| { |
| "epoch": 0.22, |
| "learning_rate": 3.6931818181818184e-05, |
| "loss": 0.0473, |
| "step": 130 |
| }, |
| { |
| "epoch": 0.24, |
| "learning_rate": 3.9772727272727275e-05, |
| "loss": 0.0253, |
| "step": 140 |
| }, |
| { |
| "epoch": 0.26, |
| "learning_rate": 4.261363636363637e-05, |
| "loss": 0.0574, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.27, |
| "learning_rate": 4.545454545454546e-05, |
| "loss": 0.0403, |
| "step": 160 |
| }, |
| { |
| "epoch": 0.29, |
| "learning_rate": 4.829545454545455e-05, |
| "loss": 0.0481, |
| "step": 170 |
| }, |
| { |
| "epoch": 0.31, |
| "learning_rate": 4.9873577749683945e-05, |
| "loss": 0.0246, |
| "step": 180 |
| }, |
| { |
| "epoch": 0.32, |
| "learning_rate": 4.955752212389381e-05, |
| "loss": 0.0094, |
| "step": 190 |
| }, |
| { |
| "epoch": 0.34, |
| "learning_rate": 4.924146649810367e-05, |
| "loss": 0.0413, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.36, |
| "learning_rate": 4.892541087231353e-05, |
| "loss": 0.0339, |
| "step": 210 |
| }, |
| { |
| "epoch": 0.38, |
| "learning_rate": 4.860935524652339e-05, |
| "loss": 0.0523, |
| "step": 220 |
| }, |
| { |
| "epoch": 0.39, |
| "learning_rate": 4.829329962073325e-05, |
| "loss": 0.0104, |
| "step": 230 |
| }, |
| { |
| "epoch": 0.41, |
| "learning_rate": 4.797724399494311e-05, |
| "loss": 0.0309, |
| "step": 240 |
| }, |
| { |
| "epoch": 0.43, |
| "learning_rate": 4.7661188369152975e-05, |
| "loss": 0.0248, |
| "step": 250 |
| }, |
| { |
| "epoch": 0.44, |
| "learning_rate": 4.734513274336283e-05, |
| "loss": 0.0306, |
| "step": 260 |
| }, |
| { |
| "epoch": 0.46, |
| "learning_rate": 4.70290771175727e-05, |
| "loss": 0.0162, |
| "step": 270 |
| }, |
| { |
| "epoch": 0.48, |
| "learning_rate": 4.6713021491782554e-05, |
| "loss": 0.016, |
| "step": 280 |
| }, |
| { |
| "epoch": 0.49, |
| "learning_rate": 4.639696586599241e-05, |
| "loss": 0.0103, |
| "step": 290 |
| }, |
| { |
| "epoch": 0.51, |
| "learning_rate": 4.608091024020228e-05, |
| "loss": 0.0196, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.53, |
| "learning_rate": 4.576485461441214e-05, |
| "loss": 0.0264, |
| "step": 310 |
| }, |
| { |
| "epoch": 0.55, |
| "learning_rate": 4.5448798988622e-05, |
| "loss": 0.0344, |
| "step": 320 |
| }, |
| { |
| "epoch": 0.56, |
| "learning_rate": 4.5132743362831855e-05, |
| "loss": 0.036, |
| "step": 330 |
| }, |
| { |
| "epoch": 0.58, |
| "learning_rate": 4.4816687737041726e-05, |
| "loss": 0.0082, |
| "step": 340 |
| }, |
| { |
| "epoch": 0.6, |
| "learning_rate": 4.4500632111251584e-05, |
| "loss": 0.0149, |
| "step": 350 |
| }, |
| { |
| "epoch": 0.61, |
| "learning_rate": 4.418457648546144e-05, |
| "loss": 0.0082, |
| "step": 360 |
| }, |
| { |
| "epoch": 0.63, |
| "learning_rate": 4.3868520859671306e-05, |
| "loss": 0.0451, |
| "step": 370 |
| }, |
| { |
| "epoch": 0.65, |
| "learning_rate": 4.355246523388117e-05, |
| "loss": 0.0055, |
| "step": 380 |
| }, |
| { |
| "epoch": 0.67, |
| "learning_rate": 4.323640960809103e-05, |
| "loss": 0.0087, |
| "step": 390 |
| }, |
| { |
| "epoch": 0.68, |
| "learning_rate": 4.2920353982300885e-05, |
| "loss": 0.0052, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.7, |
| "learning_rate": 4.260429835651075e-05, |
| "loss": 0.0033, |
| "step": 410 |
| }, |
| { |
| "epoch": 0.72, |
| "learning_rate": 4.2288242730720607e-05, |
| "loss": 0.0148, |
| "step": 420 |
| }, |
| { |
| "epoch": 0.73, |
| "learning_rate": 4.197218710493047e-05, |
| "loss": 0.0134, |
| "step": 430 |
| }, |
| { |
| "epoch": 0.75, |
| "learning_rate": 4.165613147914033e-05, |
| "loss": 0.0125, |
| "step": 440 |
| }, |
| { |
| "epoch": 0.77, |
| "learning_rate": 4.134007585335019e-05, |
| "loss": 0.0531, |
| "step": 450 |
| }, |
| { |
| "epoch": 0.78, |
| "learning_rate": 4.102402022756005e-05, |
| "loss": 0.0059, |
| "step": 460 |
| }, |
| { |
| "epoch": 0.8, |
| "learning_rate": 4.0707964601769914e-05, |
| "loss": 0.0059, |
| "step": 470 |
| }, |
| { |
| "epoch": 0.82, |
| "learning_rate": 4.039190897597978e-05, |
| "loss": 0.0194, |
| "step": 480 |
| }, |
| { |
| "epoch": 0.84, |
| "learning_rate": 4.0075853350189636e-05, |
| "loss": 0.0111, |
| "step": 490 |
| }, |
| { |
| "epoch": 0.85, |
| "learning_rate": 3.9759797724399494e-05, |
| "loss": 0.0321, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.87, |
| "learning_rate": 3.944374209860936e-05, |
| "loss": 0.0261, |
| "step": 510 |
| }, |
| { |
| "epoch": 0.89, |
| "learning_rate": 3.912768647281922e-05, |
| "loss": 0.0196, |
| "step": 520 |
| }, |
| { |
| "epoch": 0.9, |
| "learning_rate": 3.881163084702908e-05, |
| "loss": 0.0038, |
| "step": 530 |
| }, |
| { |
| "epoch": 0.92, |
| "learning_rate": 3.849557522123894e-05, |
| "loss": 0.0216, |
| "step": 540 |
| }, |
| { |
| "epoch": 0.94, |
| "learning_rate": 3.81795195954488e-05, |
| "loss": 0.0047, |
| "step": 550 |
| }, |
| { |
| "epoch": 0.96, |
| "learning_rate": 3.7863463969658666e-05, |
| "loss": 0.0095, |
| "step": 560 |
| }, |
| { |
| "epoch": 0.97, |
| "learning_rate": 3.754740834386852e-05, |
| "loss": 0.0036, |
| "step": 570 |
| }, |
| { |
| "epoch": 0.99, |
| "learning_rate": 3.723135271807838e-05, |
| "loss": 0.0025, |
| "step": 580 |
| }, |
| { |
| "epoch": 1.0, |
| "eval_accuracy": 0.996, |
| "eval_loss": 0.015431854873895645, |
| "eval_runtime": 44.7221, |
| "eval_samples_per_second": 44.721, |
| "eval_steps_per_second": 2.795, |
| "step": 586 |
| } |
| ], |
| "logging_steps": 10, |
| "max_steps": 1758, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 3, |
| "save_steps": 500, |
| "total_flos": 2.9390551738299924e+18, |
| "train_batch_size": 16, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
