{ "best_metric": 0.2431316375732422, "best_model_checkpoint": "google/vit-base-patch16-224-in21k_Simpsons_Family_Members/checkpoint-1119", "epoch": 3.0, "global_step": 1119, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "learning_rate": 0.0001998212689901698, "loss": 1.5773, "step": 1 }, { "epoch": 1.0, "eval_Macro F1": 0.7261198751167324, "eval_Macro Precision": 0.8922355318227739, "eval_Macro Recall": 0.7777777777777778, "eval_Micro F1": 0.7772277227722773, "eval_Micro Precision": 0.7772277227722773, "eval_Micro Recall": 0.7772277227722773, "eval_Weighted F1": 0.7263266810424557, "eval_Weighted Precision": 0.8932723208097979, "eval_Weighted Recall": 0.7772277227722773, "eval_accuracy": 0.7772277227722773, "eval_loss": 1.0481945276260376, "eval_runtime": 363.654, "eval_samples_per_second": 1.111, "eval_steps_per_second": 0.14, "step": 373 }, { "epoch": 1.34, "learning_rate": 0.00011063449508489723, "loss": 0.1598, "step": 500 }, { "epoch": 2.0, "eval_Macro F1": 0.9025589061327131, "eval_Macro Precision": 0.921938438296667, "eval_Macro Recall": 0.9060185185185186, "eval_Micro F1": 0.905940594059406, "eval_Micro Precision": 0.905940594059406, "eval_Micro Recall": 0.905940594059406, "eval_Weighted F1": 0.9027529236623092, "eval_Weighted Precision": 0.9224004291893639, "eval_Weighted Recall": 0.905940594059406, "eval_accuracy": 0.905940594059406, "eval_loss": 0.3902488946914673, "eval_runtime": 350.3954, "eval_samples_per_second": 1.153, "eval_steps_per_second": 0.146, "step": 746 }, { "epoch": 2.68, "learning_rate": 2.126899016979446e-05, "loss": 0.027, "step": 1000 }, { "epoch": 3.0, "eval_Macro F1": 0.9521097000368888, "eval_Macro Precision": 0.9601291764998138, "eval_Macro Recall": 0.9530864197530864, "eval_Micro F1": 0.9529702970297029, "eval_Micro Precision": 0.9529702970297029, "eval_Micro Recall": 0.9529702970297029, "eval_Weighted F1": 0.9522288953284159, "eval_Weighted Precision": 0.9604642926098597, "eval_Weighted Recall": 0.9529702970297029, "eval_accuracy": 0.9529702970297029, "eval_loss": 0.2431316375732422, "eval_runtime": 360.7329, "eval_samples_per_second": 1.12, "eval_steps_per_second": 0.141, "step": 1119 }, { "epoch": 3.0, "step": 1119, "total_flos": 1.3846641003515658e+18, "train_loss": 0.08599743702456394, "train_runtime": 48467.025, "train_samples_per_second": 0.369, "train_steps_per_second": 0.023 } ], "max_steps": 1119, "num_train_epochs": 3, "total_flos": 1.3846641003515658e+18, "trial_name": null, "trial_params": null }