| { | |
| "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 | |
| } | |