{ "best_metric": 0.8818666666666667, "best_model_checkpoint": "/opt/ml/model/checkpoint-3987", "epoch": 4.0, "global_step": 5316, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.1, "learning_rate": 7.92e-06, "loss": 0.6945, "step": 132 }, { "epoch": 0.2, "learning_rate": 1.578e-05, "loss": 0.6107, "step": 264 }, { "epoch": 0.3, "learning_rate": 2.37e-05, "loss": 0.5082, "step": 396 }, { "epoch": 0.4, "learning_rate": 2.983181063122924e-05, "loss": 0.4632, "step": 528 }, { "epoch": 0.5, "learning_rate": 2.9009551495016612e-05, "loss": 0.419, "step": 660 }, { "epoch": 0.6, "learning_rate": 2.8187292358803985e-05, "loss": 0.4142, "step": 792 }, { "epoch": 0.7, "learning_rate": 2.736503322259136e-05, "loss": 0.3748, "step": 924 }, { "epoch": 0.79, "learning_rate": 2.6542774086378738e-05, "loss": 0.3793, "step": 1056 }, { "epoch": 0.89, "learning_rate": 2.5720514950166114e-05, "loss": 0.3584, "step": 1188 }, { "epoch": 0.99, "learning_rate": 2.489825581395349e-05, "loss": 0.3456, "step": 1320 }, { "epoch": 1.0, "eval_accuracy": 0.8576, "eval_f1": 0.8572192513368984, "eval_loss": 0.32413816452026367, "eval_precision": 0.8778751369112815, "eval_recall": 0.8375130616509927, "eval_runtime": 107.1381, "eval_samples_per_second": 35.002, "step": 1329 }, { "epoch": 1.09, "learning_rate": 2.4075996677740867e-05, "loss": 0.2832, "step": 1452 }, { "epoch": 1.19, "learning_rate": 2.325373754152824e-05, "loss": 0.2625, "step": 1584 }, { "epoch": 1.29, "learning_rate": 2.2431478405315613e-05, "loss": 0.2583, "step": 1716 }, { "epoch": 1.39, "learning_rate": 2.160921926910299e-05, "loss": 0.264, "step": 1848 }, { "epoch": 1.49, "learning_rate": 2.0786960132890365e-05, "loss": 0.2579, "step": 1980 }, { "epoch": 1.59, "learning_rate": 1.9964700996677742e-05, "loss": 0.2426, "step": 2112 }, { "epoch": 1.69, "learning_rate": 1.9142441860465118e-05, "loss": 0.243, "step": 2244 }, { "epoch": 1.79, "learning_rate": 1.832641196013289e-05, "loss": 0.2642, "step": 2376 }, { "epoch": 1.89, "learning_rate": 1.7504152823920267e-05, "loss": 0.2632, "step": 2508 }, { "epoch": 1.99, "learning_rate": 1.6681893687707643e-05, "loss": 0.2564, "step": 2640 }, { "epoch": 2.0, "eval_accuracy": 0.8696, "eval_f1": 0.8666484865012271, "eval_loss": 0.3016686737537384, "eval_precision": 0.9064460924130062, "eval_recall": 0.8301985370950888, "eval_runtime": 107.0545, "eval_samples_per_second": 35.029, "step": 2658 }, { "epoch": 2.09, "learning_rate": 1.5859634551495016e-05, "loss": 0.1767, "step": 2772 }, { "epoch": 2.19, "learning_rate": 1.504360465116279e-05, "loss": 0.1527, "step": 2904 }, { "epoch": 2.28, "learning_rate": 1.4221345514950167e-05, "loss": 0.1381, "step": 3036 }, { "epoch": 2.38, "learning_rate": 1.3399086378737542e-05, "loss": 0.1516, "step": 3168 }, { "epoch": 2.48, "learning_rate": 1.2576827242524916e-05, "loss": 0.1508, "step": 3300 }, { "epoch": 2.58, "learning_rate": 1.1754568106312293e-05, "loss": 0.1479, "step": 3432 }, { "epoch": 2.68, "learning_rate": 1.093230897009967e-05, "loss": 0.1341, "step": 3564 }, { "epoch": 2.78, "learning_rate": 1.0110049833887042e-05, "loss": 0.1474, "step": 3696 }, { "epoch": 2.88, "learning_rate": 9.287790697674419e-06, "loss": 0.1462, "step": 3828 }, { "epoch": 2.98, "learning_rate": 8.465531561461795e-06, "loss": 0.1444, "step": 3960 }, { "epoch": 3.0, "eval_accuracy": 0.8818666666666667, "eval_f1": 0.8856774193548387, "eval_loss": 0.37261688709259033, "eval_precision": 0.8750637429882713, "eval_recall": 0.896551724137931, "eval_runtime": 107.1711, "eval_samples_per_second": 34.991, "step": 3987 }, { "epoch": 3.08, "learning_rate": 7.64327242524917e-06, "loss": 0.0754, "step": 4092 }, { "epoch": 3.18, "learning_rate": 6.821013289036545e-06, "loss": 0.0811, "step": 4224 }, { "epoch": 3.28, "learning_rate": 5.9987541528239206e-06, "loss": 0.0808, "step": 4356 }, { "epoch": 3.38, "learning_rate": 5.176495016611296e-06, "loss": 0.0752, "step": 4488 }, { "epoch": 3.48, "learning_rate": 4.354235880398672e-06, "loss": 0.0803, "step": 4620 }, { "epoch": 3.58, "learning_rate": 3.5319767441860467e-06, "loss": 0.0734, "step": 4752 }, { "epoch": 3.67, "learning_rate": 2.7097176079734222e-06, "loss": 0.0615, "step": 4884 }, { "epoch": 3.77, "learning_rate": 1.8874584717607975e-06, "loss": 0.0605, "step": 5016 }, { "epoch": 3.87, "learning_rate": 1.0651993355481726e-06, "loss": 0.0781, "step": 5148 }, { "epoch": 3.97, "learning_rate": 2.491694352159469e-07, "loss": 0.0713, "step": 5280 }, { "epoch": 4.0, "eval_accuracy": 0.8818666666666667, "eval_f1": 0.8882723833543505, "eval_loss": 0.5432285666465759, "eval_precision": 0.8586055582642613, "eval_recall": 0.9200626959247649, "eval_runtime": 107.1677, "eval_samples_per_second": 34.992, "step": 5316 } ], "max_steps": 5316, "num_train_epochs": 4, "total_flos": 8223951552000000.0, "trial_name": null, "trial_params": null }