| { | |
| "best_metric": 0.04133550822734833, | |
| "best_model_checkpoint": "data/models/chemical/final/checkpoint-20571", | |
| "epoch": 1.0, | |
| "global_step": 20571, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.02, | |
| "learning_rate": 9.922807350153128e-05, | |
| "loss": 0.094, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 9.675614700306256e-05, | |
| "loss": 0.0704, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "learning_rate": 9.428422050459385e-05, | |
| "loss": 0.0626, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 9.181229400612513e-05, | |
| "loss": 0.0621, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 8.934036750765641e-05, | |
| "loss": 0.0586, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "learning_rate": 8.68684410091877e-05, | |
| "loss": 0.0535, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "learning_rate": 8.439651451071898e-05, | |
| "loss": 0.0553, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 8.192458801225026e-05, | |
| "loss": 0.0514, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 7.945266151378155e-05, | |
| "loss": 0.0513, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 7.698073501531283e-05, | |
| "loss": 0.0497, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 7.45088085168441e-05, | |
| "loss": 0.0466, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 7.203688201837538e-05, | |
| "loss": 0.0481, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 6.956495551990666e-05, | |
| "loss": 0.0433, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "learning_rate": 6.709302902143795e-05, | |
| "loss": 0.0442, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 6.462110252296923e-05, | |
| "loss": 0.0471, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 6.214917602450051e-05, | |
| "loss": 0.0451, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.41, | |
| "learning_rate": 5.96772495260318e-05, | |
| "loss": 0.0428, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.44, | |
| "learning_rate": 5.720532302756308e-05, | |
| "loss": 0.0406, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.46, | |
| "learning_rate": 5.473339652909436e-05, | |
| "loss": 0.0448, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "learning_rate": 5.226147003062564e-05, | |
| "loss": 0.0413, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "learning_rate": 4.9789543532156925e-05, | |
| "loss": 0.0408, | |
| "step": 10500 | |
| }, | |
| { | |
| "epoch": 0.53, | |
| "learning_rate": 4.7317617033688203e-05, | |
| "loss": 0.0386, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 0.56, | |
| "learning_rate": 4.484569053521948e-05, | |
| "loss": 0.0402, | |
| "step": 11500 | |
| }, | |
| { | |
| "epoch": 0.58, | |
| "learning_rate": 4.2373764036750766e-05, | |
| "loss": 0.0365, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "learning_rate": 3.990183753828205e-05, | |
| "loss": 0.038, | |
| "step": 12500 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "learning_rate": 3.742991103981333e-05, | |
| "loss": 0.0346, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 0.66, | |
| "learning_rate": 3.4957984541344614e-05, | |
| "loss": 0.0356, | |
| "step": 13500 | |
| }, | |
| { | |
| "epoch": 0.68, | |
| "learning_rate": 3.248605804287589e-05, | |
| "loss": 0.0346, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 3.0014131544407173e-05, | |
| "loss": 0.0304, | |
| "step": 14500 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "learning_rate": 2.754220504593846e-05, | |
| "loss": 0.0334, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "learning_rate": 2.5070278547469743e-05, | |
| "loss": 0.0312, | |
| "step": 15500 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "learning_rate": 2.259835204900102e-05, | |
| "loss": 0.0333, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "learning_rate": 2.0126425550532302e-05, | |
| "loss": 0.0321, | |
| "step": 16500 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 1.7654499052063587e-05, | |
| "loss": 0.0338, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "learning_rate": 1.5182572553594867e-05, | |
| "loss": 0.0309, | |
| "step": 17500 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "learning_rate": 1.271064605512615e-05, | |
| "loss": 0.0295, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "learning_rate": 1.023871955665743e-05, | |
| "loss": 0.0281, | |
| "step": 18500 | |
| }, | |
| { | |
| "epoch": 0.92, | |
| "learning_rate": 7.766793058188713e-06, | |
| "loss": 0.0303, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "learning_rate": 5.294866559719994e-06, | |
| "loss": 0.0294, | |
| "step": 19500 | |
| }, | |
| { | |
| "epoch": 0.97, | |
| "learning_rate": 2.822940061251276e-06, | |
| "loss": 0.0274, | |
| "step": 20000 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "learning_rate": 3.51013562782558e-07, | |
| "loss": 0.0283, | |
| "step": 20500 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.9787234042553191, | |
| "eval_f1": 0.9333333333333333, | |
| "eval_loss": 0.04133550822734833, | |
| "eval_precision": 1.0, | |
| "eval_recall": 0.875, | |
| "eval_runtime": 5.3542, | |
| "eval_samples_per_second": 0.747, | |
| "eval_steps_per_second": 0.187, | |
| "step": 20571 | |
| } | |
| ], | |
| "max_steps": 20571, | |
| "num_train_epochs": 1, | |
| "total_flos": 8.6000072575104e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |