{ "best_metric": 0.0017420644871890545, "best_model_checkpoint": "./vit-base-beans-demo-v5/checkpoint-400", "epoch": 1.0, "global_step": 441, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.02, "learning_rate": 0.00019546485260770976, "loss": 0.273, "step": 10 }, { "epoch": 0.05, "learning_rate": 0.0001909297052154195, "loss": 0.0234, "step": 20 }, { "epoch": 0.07, "learning_rate": 0.00018639455782312926, "loss": 0.021, "step": 30 }, { "epoch": 0.09, "learning_rate": 0.000181859410430839, "loss": 0.0173, "step": 40 }, { "epoch": 0.11, "learning_rate": 0.00017732426303854876, "loss": 0.0073, "step": 50 }, { "epoch": 0.14, "learning_rate": 0.0001727891156462585, "loss": 0.0081, "step": 60 }, { "epoch": 0.16, "learning_rate": 0.00016825396825396826, "loss": 0.0219, "step": 70 }, { "epoch": 0.18, "learning_rate": 0.000163718820861678, "loss": 0.004, "step": 80 }, { "epoch": 0.2, "learning_rate": 0.00015918367346938776, "loss": 0.0039, "step": 90 }, { "epoch": 0.23, "learning_rate": 0.00015464852607709753, "loss": 0.0033, "step": 100 }, { "epoch": 0.23, "eval_f1": 1.0, "eval_loss": 0.0031821790616959333, "eval_runtime": 31.2658, "eval_samples_per_second": 36.781, "eval_steps_per_second": 4.606, "step": 100 }, { "epoch": 0.25, "learning_rate": 0.00015011337868480726, "loss": 0.003, "step": 110 }, { "epoch": 0.27, "learning_rate": 0.000145578231292517, "loss": 0.0027, "step": 120 }, { "epoch": 0.29, "learning_rate": 0.00014104308390022676, "loss": 0.0024, "step": 130 }, { "epoch": 0.32, "learning_rate": 0.0001365079365079365, "loss": 0.0026, "step": 140 }, { "epoch": 0.34, "learning_rate": 0.00013197278911564626, "loss": 0.0022, "step": 150 }, { "epoch": 0.36, "learning_rate": 0.000127437641723356, "loss": 0.002, "step": 160 }, { "epoch": 0.39, "learning_rate": 0.00012290249433106578, "loss": 0.0019, "step": 170 }, { "epoch": 0.41, "learning_rate": 0.00011836734693877552, "loss": 0.0021, "step": 180 }, { "epoch": 0.43, "learning_rate": 0.00011383219954648527, "loss": 0.0017, "step": 190 }, { "epoch": 0.45, "learning_rate": 0.000109297052154195, "loss": 0.0017, "step": 200 }, { "epoch": 0.45, "eval_f1": 1.0, "eval_loss": 0.0017858472419902682, "eval_runtime": 30.1491, "eval_samples_per_second": 38.144, "eval_steps_per_second": 4.776, "step": 200 }, { "epoch": 0.48, "learning_rate": 0.00010476190476190477, "loss": 0.0016, "step": 210 }, { "epoch": 0.5, "learning_rate": 0.0001002267573696145, "loss": 0.0015, "step": 220 }, { "epoch": 0.52, "learning_rate": 9.569160997732427e-05, "loss": 0.002, "step": 230 }, { "epoch": 0.54, "learning_rate": 9.115646258503402e-05, "loss": 0.0014, "step": 240 }, { "epoch": 0.57, "learning_rate": 8.662131519274377e-05, "loss": 0.0013, "step": 250 }, { "epoch": 0.59, "learning_rate": 8.208616780045352e-05, "loss": 0.0013, "step": 260 }, { "epoch": 0.61, "learning_rate": 7.755102040816327e-05, "loss": 0.0013, "step": 270 }, { "epoch": 0.63, "learning_rate": 7.301587301587302e-05, "loss": 0.0012, "step": 280 }, { "epoch": 0.66, "learning_rate": 6.848072562358277e-05, "loss": 0.0012, "step": 290 }, { "epoch": 0.68, "learning_rate": 6.394557823129253e-05, "loss": 0.0012, "step": 300 }, { "epoch": 0.68, "eval_f1": 0.9990009990009989, "eval_loss": 0.001951999613083899, "eval_runtime": 30.3102, "eval_samples_per_second": 37.941, "eval_steps_per_second": 4.751, "step": 300 }, { "epoch": 0.7, "learning_rate": 5.9410430839002274e-05, "loss": 0.0011, "step": 310 }, { "epoch": 0.73, "learning_rate": 5.4875283446712023e-05, "loss": 0.0011, "step": 320 }, { "epoch": 0.75, "learning_rate": 5.034013605442177e-05, "loss": 0.0011, "step": 330 }, { "epoch": 0.77, "learning_rate": 4.580498866213152e-05, "loss": 0.0011, "step": 340 }, { "epoch": 0.79, "learning_rate": 4.126984126984127e-05, "loss": 0.001, "step": 350 }, { "epoch": 0.82, "learning_rate": 3.673469387755102e-05, "loss": 0.001, "step": 360 }, { "epoch": 0.84, "learning_rate": 3.219954648526077e-05, "loss": 0.001, "step": 370 }, { "epoch": 0.86, "learning_rate": 2.7664399092970522e-05, "loss": 0.001, "step": 380 }, { "epoch": 0.88, "learning_rate": 2.3129251700680275e-05, "loss": 0.001, "step": 390 }, { "epoch": 0.91, "learning_rate": 1.8594104308390022e-05, "loss": 0.001, "step": 400 }, { "epoch": 0.91, "eval_f1": 0.9990009990009989, "eval_loss": 0.0017420644871890545, "eval_runtime": 30.1934, "eval_samples_per_second": 38.088, "eval_steps_per_second": 4.769, "step": 400 }, { "epoch": 0.93, "learning_rate": 1.4058956916099775e-05, "loss": 0.0009, "step": 410 }, { "epoch": 0.95, "learning_rate": 9.523809523809523e-06, "loss": 0.0009, "step": 420 }, { "epoch": 0.98, "learning_rate": 4.9886621315192745e-06, "loss": 0.001, "step": 430 }, { "epoch": 1.0, "learning_rate": 4.53514739229025e-07, "loss": 0.001, "step": 440 }, { "epoch": 1.0, "step": 441, "total_flos": 5.463185267828736e+17, "train_loss": 0.009828982074758727, "train_runtime": 378.8692, "train_samples_per_second": 18.608, "train_steps_per_second": 1.164 } ], "max_steps": 441, "num_train_epochs": 1, "total_flos": 5.463185267828736e+17, "trial_name": null, "trial_params": null }