| { | |
| "best_metric": 0.7126299142837524, | |
| "best_model_checkpoint": "./vit-base/checkpoint-200", | |
| "epoch": 20.0, | |
| "global_step": 280, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.71, | |
| "learning_rate": 0.00019357142857142858, | |
| "loss": 2.073, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 1.43, | |
| "learning_rate": 0.00018642857142857144, | |
| "loss": 0.6674, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 2.14, | |
| "learning_rate": 0.0001792857142857143, | |
| "loss": 0.1817, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 2.86, | |
| "learning_rate": 0.00017214285714285717, | |
| "loss": 0.0578, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 3.57, | |
| "learning_rate": 0.000165, | |
| "loss": 0.0041, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 3.57, | |
| "eval_accuracy": 0.7575757575757576, | |
| "eval_loss": 0.5183216333389282, | |
| "eval_runtime": 1.6943, | |
| "eval_samples_per_second": 19.477, | |
| "eval_steps_per_second": 2.951, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 4.29, | |
| "learning_rate": 0.00015785714285714285, | |
| "loss": 0.007, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "learning_rate": 0.0001507142857142857, | |
| "loss": 0.0018, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 5.71, | |
| "learning_rate": 0.0001435714285714286, | |
| "loss": 0.0009, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 6.43, | |
| "learning_rate": 0.00013642857142857144, | |
| "loss": 0.0007, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 7.14, | |
| "learning_rate": 0.0001292857142857143, | |
| "loss": 0.0006, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 7.14, | |
| "eval_accuracy": 0.8181818181818182, | |
| "eval_loss": 0.7554123401641846, | |
| "eval_runtime": 1.4254, | |
| "eval_samples_per_second": 23.151, | |
| "eval_steps_per_second": 3.508, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 7.86, | |
| "learning_rate": 0.00012214285714285715, | |
| "loss": 0.0005, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 8.57, | |
| "learning_rate": 0.00011499999999999999, | |
| "loss": 0.0005, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 9.29, | |
| "learning_rate": 0.00010785714285714286, | |
| "loss": 0.0004, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "learning_rate": 0.00010071428571428571, | |
| "loss": 0.0004, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 10.71, | |
| "learning_rate": 9.357142857142858e-05, | |
| "loss": 0.0004, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 10.71, | |
| "eval_accuracy": 0.8181818181818182, | |
| "eval_loss": 0.7197360396385193, | |
| "eval_runtime": 1.4885, | |
| "eval_samples_per_second": 22.17, | |
| "eval_steps_per_second": 3.359, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 11.43, | |
| "learning_rate": 8.642857142857143e-05, | |
| "loss": 0.0004, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 12.14, | |
| "learning_rate": 7.928571428571429e-05, | |
| "loss": 0.0003, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 12.86, | |
| "learning_rate": 7.214285714285714e-05, | |
| "loss": 0.0003, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 13.57, | |
| "learning_rate": 6.500000000000001e-05, | |
| "loss": 0.0003, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 14.29, | |
| "learning_rate": 5.785714285714287e-05, | |
| "loss": 0.0003, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 14.29, | |
| "eval_accuracy": 0.8181818181818182, | |
| "eval_loss": 0.7126299142837524, | |
| "eval_runtime": 1.5758, | |
| "eval_samples_per_second": 20.942, | |
| "eval_steps_per_second": 3.173, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 15.0, | |
| "learning_rate": 5.0714285714285716e-05, | |
| "loss": 0.0003, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 15.71, | |
| "learning_rate": 4.3571428571428576e-05, | |
| "loss": 0.0003, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 16.43, | |
| "learning_rate": 3.642857142857143e-05, | |
| "loss": 0.0003, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 17.14, | |
| "learning_rate": 2.9285714285714288e-05, | |
| "loss": 0.0003, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 17.86, | |
| "learning_rate": 2.214285714285714e-05, | |
| "loss": 0.0003, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 17.86, | |
| "eval_accuracy": 0.8181818181818182, | |
| "eval_loss": 0.7098276615142822, | |
| "eval_runtime": 1.5439, | |
| "eval_samples_per_second": 21.374, | |
| "eval_steps_per_second": 3.239, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 18.57, | |
| "learning_rate": 1.5e-05, | |
| "loss": 0.0003, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 19.29, | |
| "learning_rate": 7.857142857142858e-06, | |
| "loss": 0.0003, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "learning_rate": 7.142857142857143e-07, | |
| "loss": 0.0003, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "step": 280, | |
| "total_flos": 8.5248064484352e+16, | |
| "train_loss": 0.1071862239856273, | |
| "train_runtime": 140.5527, | |
| "train_samples_per_second": 7.826, | |
| "train_steps_per_second": 1.992 | |
| } | |
| ], | |
| "max_steps": 280, | |
| "num_train_epochs": 20, | |
| "total_flos": 8.5248064484352e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |