| { |
| "best_metric": 0.98175456135966, |
| "best_model_checkpoint": "vit-base-patch16-224-Futurama_Image_multilabel_clf/checkpoint-4580", |
| "epoch": 8.0, |
| "global_step": 7328, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.0, |
| "learning_rate": 1.999727074235808e-05, |
| "loss": 0.6884, |
| "step": 1 |
| }, |
| { |
| "epoch": 0.55, |
| "learning_rate": 1.8635371179039304e-05, |
| "loss": 0.2456, |
| "step": 500 |
| }, |
| { |
| "epoch": 1.0, |
| "eval_accuracy": 0.9481441048034934, |
| "eval_f1": 0.9710691823899372, |
| "eval_loss": 0.07229584455490112, |
| "eval_roc_auc": 0.9745990820070386, |
| "eval_runtime": 1680.1374, |
| "eval_samples_per_second": 1.09, |
| "eval_steps_per_second": 0.136, |
| "step": 916 |
| }, |
| { |
| "epoch": 1.09, |
| "learning_rate": 1.7270742358078607e-05, |
| "loss": 0.0749, |
| "step": 1000 |
| }, |
| { |
| "epoch": 1.64, |
| "learning_rate": 1.5906113537117906e-05, |
| "loss": 0.0269, |
| "step": 1500 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_accuracy": 0.9639737991266376, |
| "eval_f1": 0.9799095931692616, |
| "eval_loss": 0.05449194833636284, |
| "eval_roc_auc": 0.9818209165245149, |
| "eval_runtime": 1758.0937, |
| "eval_samples_per_second": 1.042, |
| "eval_steps_per_second": 0.13, |
| "step": 1832 |
| }, |
| { |
| "epoch": 2.18, |
| "learning_rate": 1.4541484716157206e-05, |
| "loss": 0.0234, |
| "step": 2000 |
| }, |
| { |
| "epoch": 2.73, |
| "learning_rate": 1.3176855895196507e-05, |
| "loss": 0.0086, |
| "step": 2500 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_accuracy": 0.9623362445414847, |
| "eval_f1": 0.9794073329984931, |
| "eval_loss": 0.05799925699830055, |
| "eval_roc_auc": 0.9814291406807462, |
| "eval_runtime": 1649.1127, |
| "eval_samples_per_second": 1.111, |
| "eval_steps_per_second": 0.139, |
| "step": 2748 |
| }, |
| { |
| "epoch": 3.28, |
| "learning_rate": 1.181222707423581e-05, |
| "loss": 0.0046, |
| "step": 3000 |
| }, |
| { |
| "epoch": 3.82, |
| "learning_rate": 1.044759825327511e-05, |
| "loss": 0.0044, |
| "step": 3500 |
| }, |
| { |
| "epoch": 4.0, |
| "eval_accuracy": 0.9650655021834061, |
| "eval_f1": 0.9814350225790265, |
| "eval_loss": 0.06122291088104248, |
| "eval_roc_auc": 0.983205248199549, |
| "eval_runtime": 1658.3164, |
| "eval_samples_per_second": 1.105, |
| "eval_steps_per_second": 0.138, |
| "step": 3664 |
| }, |
| { |
| "epoch": 4.37, |
| "learning_rate": 9.082969432314411e-06, |
| "loss": 0.0022, |
| "step": 4000 |
| }, |
| { |
| "epoch": 4.91, |
| "learning_rate": 7.718340611353714e-06, |
| "loss": 0.0027, |
| "step": 4500 |
| }, |
| { |
| "epoch": 5.0, |
| "eval_accuracy": 0.9672489082969432, |
| "eval_f1": 0.98175456135966, |
| "eval_loss": 0.05915337800979614, |
| "eval_roc_auc": 0.984184901660414, |
| "eval_runtime": 1650.3185, |
| "eval_samples_per_second": 1.11, |
| "eval_steps_per_second": 0.139, |
| "step": 4580 |
| }, |
| { |
| "epoch": 5.46, |
| "learning_rate": 6.353711790393014e-06, |
| "loss": 0.0017, |
| "step": 5000 |
| }, |
| { |
| "epoch": 6.0, |
| "eval_accuracy": 0.9645196506550219, |
| "eval_f1": 0.9800399201596807, |
| "eval_loss": 0.06344735622406006, |
| "eval_roc_auc": 0.9831794434587476, |
| "eval_runtime": 1655.6208, |
| "eval_samples_per_second": 1.107, |
| "eval_steps_per_second": 0.138, |
| "step": 5496 |
| }, |
| { |
| "epoch": 6.0, |
| "learning_rate": 4.989082969432315e-06, |
| "loss": 0.0013, |
| "step": 5500 |
| }, |
| { |
| "epoch": 6.55, |
| "learning_rate": 3.624454148471616e-06, |
| "loss": 0.0012, |
| "step": 6000 |
| }, |
| { |
| "epoch": 7.0, |
| "eval_accuracy": 0.9667030567685589, |
| "eval_f1": 0.98173630222667, |
| "eval_loss": 0.06574680656194687, |
| "eval_roc_auc": 0.9839758975166857, |
| "eval_runtime": 1588.6633, |
| "eval_samples_per_second": 1.153, |
| "eval_steps_per_second": 0.144, |
| "step": 6412 |
| }, |
| { |
| "epoch": 7.1, |
| "learning_rate": 2.259825327510917e-06, |
| "loss": 0.0006, |
| "step": 6500 |
| }, |
| { |
| "epoch": 7.64, |
| "learning_rate": 8.951965065502185e-07, |
| "loss": 0.0005, |
| "step": 7000 |
| }, |
| { |
| "epoch": 8.0, |
| "eval_accuracy": 0.9667030567685589, |
| "eval_f1": 0.981235926945209, |
| "eval_loss": 0.06680033355951309, |
| "eval_roc_auc": 0.9835841216729171, |
| "eval_runtime": 1637.4794, |
| "eval_samples_per_second": 1.119, |
| "eval_steps_per_second": 0.14, |
| "step": 7328 |
| }, |
| { |
| "epoch": 8.0, |
| "step": 7328, |
| "total_flos": 4.540451349762736e+18, |
| "train_loss": 0.027292877175401912, |
| "train_runtime": 171163.2402, |
| "train_samples_per_second": 0.342, |
| "train_steps_per_second": 0.043 |
| } |
| ], |
| "max_steps": 7328, |
| "num_train_epochs": 8, |
| "total_flos": 4.540451349762736e+18, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
