{ "best_metric": 0.9340639641016528, "best_model_checkpoint": "google/vit-base-patch16-224-in21k-vegetables_classification/checkpoint-1011", "epoch": 3.0, "global_step": 1011, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "learning_rate": 0.00019980217606330368, "loss": 2.4333, "step": 1 }, { "epoch": 1.0, "eval_Macro F1": 0.9080099799398524, "eval_Macro Precision": 0.9234365278821094, "eval_Macro Recall": 0.9002144522770077, "eval_Micro F1": 0.9035608308605341, "eval_Micro Precision": 0.9035608308605341, "eval_Micro Recall": 0.9035608308605341, "eval_Weighted F1": 0.9027625888607032, "eval_Weighted Precision": 0.9088281264284735, "eval_Weighted Recall": 0.9035608308605341, "eval_accuracy": 0.9035608308605341, "eval_loss": 0.3374052047729492, "eval_runtime": 1187.6363, "eval_samples_per_second": 1.135, "eval_steps_per_second": 0.142, "step": 337 }, { "epoch": 1.48, "learning_rate": 0.00010108803165182987, "loss": 0.4422, "step": 500 }, { "epoch": 2.0, "eval_Macro F1": 0.9284751708493545, "eval_Macro Precision": 0.9317762555447896, "eval_Macro Recall": 0.9273066824708632, "eval_Micro F1": 0.9228486646884273, "eval_Micro Precision": 0.9228486646884273, "eval_Micro Recall": 0.9228486646884273, "eval_Weighted F1": 0.922553414641916, "eval_Weighted Precision": 0.9248209386939668, "eval_Weighted Recall": 0.9228486646884273, "eval_accuracy": 0.9228486646884273, "eval_loss": 0.2503892481327057, "eval_runtime": 1168.2307, "eval_samples_per_second": 1.154, "eval_steps_per_second": 0.145, "step": 674 }, { "epoch": 2.97, "learning_rate": 2.1760633036597427e-06, "loss": 0.1051, "step": 1000 }, { "epoch": 3.0, "eval_Macro F1": 0.9372182266715665, "eval_Macro Precision": 0.9397527643161172, "eval_Macro Recall": 0.9353770361300282, "eval_Micro F1": 0.9339762611275965, "eval_Micro Precision": 0.9339762611275965, "eval_Micro Recall": 0.9339762611275965, "eval_Weighted F1": 0.9340639641016528, "eval_Weighted Precision": 0.9346802003761767, "eval_Weighted Recall": 0.9339762611275965, "eval_accuracy": 0.9339762611275965, "eval_loss": 0.2254980355501175, "eval_runtime": 1168.0362, "eval_samples_per_second": 1.154, "eval_steps_per_second": 0.145, "step": 1011 }, { "epoch": 3.0, "step": 1011, "total_flos": 1.2529140531986534e+18, "train_loss": 0.2735011851162632, "train_runtime": 45270.4792, "train_samples_per_second": 0.357, "train_steps_per_second": 0.022 } ], "max_steps": 1011, "num_train_epochs": 3, "total_flos": 1.2529140531986534e+18, "trial_name": null, "trial_params": null }