{ "best_metric": 0.036450520157814026, "best_model_checkpoint": "./vit-base-beans/checkpoint-280", "epoch": 1.8666666666666667, "global_step": 280, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.07, "learning_rate": 0.00019333333333333333, "loss": 0.5095, "step": 10 }, { "epoch": 0.13, "learning_rate": 0.0001866666666666667, "loss": 0.1264, "step": 20 }, { "epoch": 0.2, "learning_rate": 0.00018, "loss": 0.2591, "step": 30 }, { "epoch": 0.27, "learning_rate": 0.00017333333333333334, "loss": 0.2432, "step": 40 }, { "epoch": 0.27, "eval_accuracy": 0.96, "eval_loss": 0.1231376901268959, "eval_runtime": 6.2617, "eval_samples_per_second": 47.911, "eval_steps_per_second": 6.069, "step": 40 }, { "epoch": 0.33, "learning_rate": 0.0001666666666666667, "loss": 0.0895, "step": 50 }, { "epoch": 0.4, "learning_rate": 0.00016, "loss": 0.0498, "step": 60 }, { "epoch": 0.47, "learning_rate": 0.00015333333333333334, "loss": 0.212, "step": 70 }, { "epoch": 0.53, "learning_rate": 0.00014666666666666666, "loss": 0.074, "step": 80 }, { "epoch": 0.53, "eval_accuracy": 0.9833333333333333, "eval_loss": 0.07364834100008011, "eval_runtime": 5.1068, "eval_samples_per_second": 58.745, "eval_steps_per_second": 7.441, "step": 80 }, { "epoch": 0.6, "learning_rate": 0.00014, "loss": 0.0992, "step": 90 }, { "epoch": 0.67, "learning_rate": 0.00013333333333333334, "loss": 0.0764, "step": 100 }, { "epoch": 0.73, "learning_rate": 0.00012666666666666666, "loss": 0.0284, "step": 110 }, { "epoch": 0.8, "learning_rate": 0.00012, "loss": 0.0787, "step": 120 }, { "epoch": 0.8, "eval_accuracy": 0.98, "eval_loss": 0.08297358453273773, "eval_runtime": 6.2497, "eval_samples_per_second": 48.002, "eval_steps_per_second": 6.08, "step": 120 }, { "epoch": 0.87, "learning_rate": 0.00011333333333333334, "loss": 0.0096, "step": 130 }, { "epoch": 0.93, "learning_rate": 0.00010666666666666667, "loss": 0.0387, "step": 140 }, { "epoch": 1.0, "learning_rate": 0.0001, "loss": 0.0246, "step": 150 }, { "epoch": 1.07, "learning_rate": 9.333333333333334e-05, "loss": 0.0132, "step": 160 }, { "epoch": 1.07, "eval_accuracy": 0.9866666666666667, "eval_loss": 0.04826768860220909, "eval_runtime": 5.1012, "eval_samples_per_second": 58.809, "eval_steps_per_second": 7.449, "step": 160 }, { "epoch": 1.13, "learning_rate": 8.666666666666667e-05, "loss": 0.0056, "step": 170 }, { "epoch": 1.2, "learning_rate": 8e-05, "loss": 0.0379, "step": 180 }, { "epoch": 1.27, "learning_rate": 7.333333333333333e-05, "loss": 0.0277, "step": 190 }, { "epoch": 1.33, "learning_rate": 6.666666666666667e-05, "loss": 0.0305, "step": 200 }, { "epoch": 1.33, "eval_accuracy": 0.9866666666666667, "eval_loss": 0.05280878767371178, "eval_runtime": 6.211, "eval_samples_per_second": 48.301, "eval_steps_per_second": 6.118, "step": 200 }, { "epoch": 1.4, "learning_rate": 6e-05, "loss": 0.0052, "step": 210 }, { "epoch": 1.47, "learning_rate": 5.333333333333333e-05, "loss": 0.0055, "step": 220 }, { "epoch": 1.53, "learning_rate": 4.666666666666667e-05, "loss": 0.0047, "step": 230 }, { "epoch": 1.6, "learning_rate": 4e-05, "loss": 0.0784, "step": 240 }, { "epoch": 1.6, "eval_accuracy": 0.98, "eval_loss": 0.066566601395607, "eval_runtime": 5.7633, "eval_samples_per_second": 52.053, "eval_steps_per_second": 6.593, "step": 240 }, { "epoch": 1.67, "learning_rate": 3.3333333333333335e-05, "loss": 0.0294, "step": 250 }, { "epoch": 1.73, "learning_rate": 2.6666666666666667e-05, "loss": 0.036, "step": 260 }, { "epoch": 1.8, "learning_rate": 2e-05, "loss": 0.037, "step": 270 }, { "epoch": 1.87, "learning_rate": 1.3333333333333333e-05, "loss": 0.006, "step": 280 }, { "epoch": 1.87, "eval_accuracy": 0.99, "eval_loss": 0.036450520157814026, "eval_runtime": 5.101, "eval_samples_per_second": 58.812, "eval_steps_per_second": 7.45, "step": 280 } ], "max_steps": 300, "num_train_epochs": 2, "total_flos": 3.4716411347337216e+17, "trial_name": null, "trial_params": null }