|
{ |
|
"best_metric": 0.99, |
|
"best_model_checkpoint": "tknnguyen.2022_AML_A2_Q4/checkpoint-6750", |
|
"epoch": 3.0, |
|
"global_step": 6750, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 1.851851851851852e-05, |
|
"loss": 1.2124, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 1.7037037037037038e-05, |
|
"loss": 0.5279, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1.555555555555556e-05, |
|
"loss": 0.4215, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 1.4074074074074075e-05, |
|
"loss": 0.3597, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_accuracy": 0.9822, |
|
"eval_loss": 0.12298029661178589, |
|
"eval_runtime": 68.6299, |
|
"eval_samples_per_second": 72.855, |
|
"eval_steps_per_second": 18.214, |
|
"step": 2250 |
|
}, |
|
{ |
|
"epoch": 1.11, |
|
"learning_rate": 1.2592592592592593e-05, |
|
"loss": 0.3305, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 1.33, |
|
"learning_rate": 1.1111111111111113e-05, |
|
"loss": 0.3042, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 1.56, |
|
"learning_rate": 9.62962962962963e-06, |
|
"loss": 0.2858, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 1.78, |
|
"learning_rate": 8.148148148148148e-06, |
|
"loss": 0.2576, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 6.666666666666667e-06, |
|
"loss": 0.2548, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_accuracy": 0.9854, |
|
"eval_loss": 0.07198980450630188, |
|
"eval_runtime": 68.7186, |
|
"eval_samples_per_second": 72.76, |
|
"eval_steps_per_second": 18.19, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 2.22, |
|
"learning_rate": 5.185185185185185e-06, |
|
"loss": 0.2411, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 2.44, |
|
"learning_rate": 3.7037037037037037e-06, |
|
"loss": 0.2268, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 2.67, |
|
"learning_rate": 2.222222222222222e-06, |
|
"loss": 0.2267, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 2.89, |
|
"learning_rate": 7.407407407407407e-07, |
|
"loss": 0.2234, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_accuracy": 0.99, |
|
"eval_loss": 0.05685313791036606, |
|
"eval_runtime": 68.3871, |
|
"eval_samples_per_second": 73.113, |
|
"eval_steps_per_second": 18.278, |
|
"step": 6750 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"step": 6750, |
|
"total_flos": 1.046216869705728e+19, |
|
"train_loss": 0.3684066082989728, |
|
"train_runtime": 5023.6729, |
|
"train_samples_per_second": 26.873, |
|
"train_steps_per_second": 1.344 |
|
} |
|
], |
|
"max_steps": 6750, |
|
"num_train_epochs": 3, |
|
"total_flos": 1.046216869705728e+19, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|