{ "best_metric": 1.019775390625, "best_model_checkpoint": "autotrain-vit-base-2/checkpoint-18", "epoch": 9.0, "eval_steps": 500, "global_step": 18, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.5, "learning_rate": 0.0, "loss": 1.0986, "step": 1 }, { "epoch": 1.0, "learning_rate": 5e-05, "loss": 1.0986, "step": 2 }, { "epoch": 1.0, "eval_accuracy": 0.5, "eval_f1_macro": 0.26666666666666666, "eval_f1_micro": 0.5, "eval_f1_weighted": 0.4, "eval_loss": 1.0986328125, "eval_precision_macro": 0.2222222222222222, "eval_precision_micro": 0.5, "eval_precision_weighted": 0.3333333333333333, "eval_recall_macro": 0.3333333333333333, "eval_recall_micro": 0.5, "eval_recall_weighted": 0.5, "eval_runtime": 0.1241, "eval_samples_per_second": 32.227, "eval_steps_per_second": 8.057, "step": 2 }, { "epoch": 1.5, "learning_rate": 0.0001, "loss": 1.0986, "step": 3 }, { "epoch": 2.0, "learning_rate": 9.444444444444444e-05, "loss": 1.0999, "step": 4 }, { "epoch": 2.0, "eval_accuracy": 0.5, "eval_f1_macro": 0.2222222222222222, "eval_f1_micro": 0.5, "eval_f1_weighted": 0.3333333333333333, "eval_loss": 1.086669921875, "eval_precision_macro": 0.16666666666666666, "eval_precision_micro": 0.5, "eval_precision_weighted": 0.25, "eval_recall_macro": 0.3333333333333333, "eval_recall_micro": 0.5, "eval_recall_weighted": 0.5, "eval_runtime": 0.1558, "eval_samples_per_second": 25.675, "eval_steps_per_second": 6.419, "step": 4 }, { "epoch": 2.5, "learning_rate": 8.888888888888889e-05, "loss": 1.0785, "step": 5 }, { "epoch": 3.0, "learning_rate": 8.333333333333334e-05, "loss": 1.0745, "step": 6 }, { "epoch": 3.0, "eval_accuracy": 0.75, "eval_f1_macro": 0.6, "eval_f1_micro": 0.75, "eval_f1_weighted": 0.65, "eval_loss": 1.07421875, "eval_precision_macro": 0.5555555555555555, "eval_precision_micro": 0.75, "eval_precision_weighted": 0.5833333333333333, "eval_recall_macro": 0.6666666666666666, "eval_recall_micro": 0.75, "eval_recall_weighted": 0.75, "eval_runtime": 0.1106, "eval_samples_per_second": 36.166, "eval_steps_per_second": 9.042, "step": 6 }, { "epoch": 3.5, "learning_rate": 7.777777777777778e-05, "loss": 1.0503, "step": 7 }, { "epoch": 4.0, "learning_rate": 7.222222222222222e-05, "loss": 1.033, "step": 8 }, { "epoch": 4.0, "eval_accuracy": 0.75, "eval_f1_macro": 0.6, "eval_f1_micro": 0.75, "eval_f1_weighted": 0.65, "eval_loss": 1.061279296875, "eval_precision_macro": 0.5555555555555555, "eval_precision_micro": 0.75, "eval_precision_weighted": 0.5833333333333333, "eval_recall_macro": 0.6666666666666666, "eval_recall_micro": 0.75, "eval_recall_weighted": 0.75, "eval_runtime": 0.145, "eval_samples_per_second": 27.586, "eval_steps_per_second": 6.897, "step": 8 }, { "epoch": 4.5, "learning_rate": 6.666666666666667e-05, "loss": 1.0021, "step": 9 }, { "epoch": 5.0, "learning_rate": 6.666666666666667e-05, "loss": 1.0374, "step": 10 }, { "epoch": 5.0, "eval_accuracy": 0.75, "eval_f1_macro": 0.6, "eval_f1_micro": 0.75, "eval_f1_weighted": 0.65, "eval_loss": 1.05322265625, "eval_precision_macro": 0.5555555555555555, "eval_precision_micro": 0.75, "eval_precision_weighted": 0.5833333333333333, "eval_recall_macro": 0.6666666666666666, "eval_recall_micro": 0.75, "eval_recall_weighted": 0.75, "eval_runtime": 0.1047, "eval_samples_per_second": 38.192, "eval_steps_per_second": 9.548, "step": 10 }, { "epoch": 5.5, "learning_rate": 6.111111111111112e-05, "loss": 1.0021, "step": 11 }, { "epoch": 6.0, "learning_rate": 5.555555555555556e-05, "loss": 0.9912, "step": 12 }, { "epoch": 6.0, "eval_accuracy": 0.75, "eval_f1_macro": 0.6, "eval_f1_micro": 0.75, "eval_f1_weighted": 0.65, "eval_loss": 1.0400390625, "eval_precision_macro": 0.5555555555555555, "eval_precision_micro": 0.75, "eval_precision_weighted": 0.5833333333333333, "eval_recall_macro": 0.6666666666666666, "eval_recall_micro": 0.75, "eval_recall_weighted": 0.75, "eval_runtime": 0.1544, "eval_samples_per_second": 25.903, "eval_steps_per_second": 6.476, "step": 12 }, { "epoch": 6.5, "learning_rate": 5e-05, "loss": 0.969, "step": 13 }, { "epoch": 7.0, "learning_rate": 4.4444444444444447e-05, "loss": 0.9684, "step": 14 }, { "epoch": 7.0, "eval_accuracy": 0.75, "eval_f1_macro": 0.6, "eval_f1_micro": 0.75, "eval_f1_weighted": 0.65, "eval_loss": 1.031494140625, "eval_precision_macro": 0.5555555555555555, "eval_precision_micro": 0.75, "eval_precision_weighted": 0.5833333333333333, "eval_recall_macro": 0.6666666666666666, "eval_recall_micro": 0.75, "eval_recall_weighted": 0.75, "eval_runtime": 0.1058, "eval_samples_per_second": 37.811, "eval_steps_per_second": 9.453, "step": 14 }, { "epoch": 7.5, "learning_rate": 3.888888888888889e-05, "loss": 0.9493, "step": 15 }, { "epoch": 8.0, "learning_rate": 3.3333333333333335e-05, "loss": 0.9297, "step": 16 }, { "epoch": 8.0, "eval_accuracy": 0.75, "eval_f1_macro": 0.6, "eval_f1_micro": 0.75, "eval_f1_weighted": 0.65, "eval_loss": 1.024658203125, "eval_precision_macro": 0.5555555555555555, "eval_precision_micro": 0.75, "eval_precision_weighted": 0.5833333333333333, "eval_recall_macro": 0.6666666666666666, "eval_recall_micro": 0.75, "eval_recall_weighted": 0.75, "eval_runtime": 0.1145, "eval_samples_per_second": 34.936, "eval_steps_per_second": 8.734, "step": 16 }, { "epoch": 8.5, "learning_rate": 2.777777777777778e-05, "loss": 0.9229, "step": 17 }, { "epoch": 9.0, "learning_rate": 2.2222222222222223e-05, "loss": 0.9038, "step": 18 }, { "epoch": 9.0, "eval_accuracy": 0.75, "eval_f1_macro": 0.6, "eval_f1_micro": 0.75, "eval_f1_weighted": 0.65, "eval_loss": 1.019775390625, "eval_precision_macro": 0.5555555555555555, "eval_precision_micro": 0.75, "eval_precision_weighted": 0.5833333333333333, "eval_recall_macro": 0.6666666666666666, "eval_recall_micro": 0.75, "eval_recall_weighted": 0.75, "eval_runtime": 0.1741, "eval_samples_per_second": 22.98, "eval_steps_per_second": 5.745, "step": 18 } ], "logging_steps": 1, "max_steps": 20, "num_input_tokens_seen": 0, "num_train_epochs": 10, "save_steps": 500, "total_flos": 8369209888284672.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }