tajuarAkash's picture
Add fine-tuned TinyLlama model
d661bad
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.3599999999999999,
"eval_steps": 500,
"global_step": 850,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.032,
"grad_norm": 671614.75,
"learning_rate": 4.9493333333333336e-05,
"loss": 2.1669,
"step": 20
},
{
"epoch": 0.064,
"grad_norm": 41173.5,
"learning_rate": 4.896e-05,
"loss": 0.5516,
"step": 40
},
{
"epoch": 0.096,
"grad_norm": 14334.6923828125,
"learning_rate": 4.842666666666667e-05,
"loss": 0.2249,
"step": 60
},
{
"epoch": 0.128,
"grad_norm": 14753.2421875,
"learning_rate": 4.789333333333334e-05,
"loss": 0.1644,
"step": 80
},
{
"epoch": 0.16,
"grad_norm": 12797.7333984375,
"learning_rate": 4.736000000000001e-05,
"loss": 0.103,
"step": 100
},
{
"epoch": 0.192,
"grad_norm": 12743.939453125,
"learning_rate": 4.682666666666667e-05,
"loss": 0.0684,
"step": 120
},
{
"epoch": 0.224,
"grad_norm": 10378.1640625,
"learning_rate": 4.629333333333333e-05,
"loss": 0.0495,
"step": 140
},
{
"epoch": 0.256,
"grad_norm": 8000.236328125,
"learning_rate": 4.576e-05,
"loss": 0.0404,
"step": 160
},
{
"epoch": 0.288,
"grad_norm": 8669.037109375,
"learning_rate": 4.5226666666666665e-05,
"loss": 0.0356,
"step": 180
},
{
"epoch": 0.32,
"grad_norm": 8630.375,
"learning_rate": 4.4693333333333335e-05,
"loss": 0.0291,
"step": 200
},
{
"epoch": 0.352,
"grad_norm": 8318.3955078125,
"learning_rate": 4.4160000000000004e-05,
"loss": 0.0285,
"step": 220
},
{
"epoch": 0.384,
"grad_norm": 7282.09619140625,
"learning_rate": 4.3626666666666674e-05,
"loss": 0.0282,
"step": 240
},
{
"epoch": 0.416,
"grad_norm": 7333.845703125,
"learning_rate": 4.3093333333333336e-05,
"loss": 0.028,
"step": 260
},
{
"epoch": 0.448,
"grad_norm": 8852.5693359375,
"learning_rate": 4.256e-05,
"loss": 0.0312,
"step": 280
},
{
"epoch": 0.48,
"grad_norm": 6368.5595703125,
"learning_rate": 4.202666666666667e-05,
"loss": 0.0279,
"step": 300
},
{
"epoch": 0.512,
"grad_norm": 7761.8134765625,
"learning_rate": 4.149333333333333e-05,
"loss": 0.0277,
"step": 320
},
{
"epoch": 0.544,
"grad_norm": 6672.0615234375,
"learning_rate": 4.096e-05,
"loss": 0.0276,
"step": 340
},
{
"epoch": 0.576,
"grad_norm": 7834.865234375,
"learning_rate": 4.042666666666667e-05,
"loss": 0.0276,
"step": 360
},
{
"epoch": 0.608,
"grad_norm": 8973.3291015625,
"learning_rate": 3.989333333333333e-05,
"loss": 0.0303,
"step": 380
},
{
"epoch": 0.64,
"grad_norm": 5775.8837890625,
"learning_rate": 3.936e-05,
"loss": 0.0275,
"step": 400
},
{
"epoch": 0.672,
"grad_norm": 8265.2255859375,
"learning_rate": 3.8826666666666665e-05,
"loss": 0.0275,
"step": 420
},
{
"epoch": 0.704,
"grad_norm": 6458.6533203125,
"learning_rate": 3.8293333333333335e-05,
"loss": 0.0275,
"step": 440
},
{
"epoch": 0.736,
"grad_norm": 7894.96435546875,
"learning_rate": 3.776e-05,
"loss": 0.029,
"step": 460
},
{
"epoch": 0.768,
"grad_norm": 7347.64013671875,
"learning_rate": 3.7226666666666674e-05,
"loss": 0.0275,
"step": 480
},
{
"epoch": 0.8,
"grad_norm": 6285.21630859375,
"learning_rate": 3.669333333333334e-05,
"loss": 0.0275,
"step": 500
},
{
"epoch": 0.832,
"grad_norm": 8621.4580078125,
"learning_rate": 3.616e-05,
"loss": 0.0274,
"step": 520
},
{
"epoch": 0.864,
"grad_norm": 5599.4931640625,
"learning_rate": 3.562666666666667e-05,
"loss": 0.0293,
"step": 540
},
{
"epoch": 0.896,
"grad_norm": 6125.87890625,
"learning_rate": 3.509333333333333e-05,
"loss": 0.0272,
"step": 560
},
{
"epoch": 0.928,
"grad_norm": 6572.81689453125,
"learning_rate": 3.456e-05,
"loss": 0.0272,
"step": 580
},
{
"epoch": 0.96,
"grad_norm": 6079.21435546875,
"learning_rate": 3.402666666666667e-05,
"loss": 0.0298,
"step": 600
},
{
"epoch": 0.992,
"grad_norm": 7548.86474609375,
"learning_rate": 3.349333333333334e-05,
"loss": 0.0302,
"step": 620
},
{
"epoch": 1.024,
"grad_norm": 5836.685546875,
"learning_rate": 3.296e-05,
"loss": 0.0296,
"step": 640
},
{
"epoch": 1.056,
"grad_norm": 5482.8134765625,
"learning_rate": 3.2426666666666666e-05,
"loss": 0.0291,
"step": 660
},
{
"epoch": 1.088,
"grad_norm": 7803.9892578125,
"learning_rate": 3.1893333333333335e-05,
"loss": 0.0271,
"step": 680
},
{
"epoch": 1.12,
"grad_norm": 5504.576171875,
"learning_rate": 3.136e-05,
"loss": 0.0272,
"step": 700
},
{
"epoch": 1.152,
"grad_norm": 6076.79541015625,
"learning_rate": 3.082666666666667e-05,
"loss": 0.0273,
"step": 720
},
{
"epoch": 1.184,
"grad_norm": 5411.95458984375,
"learning_rate": 3.0293333333333334e-05,
"loss": 0.0297,
"step": 740
},
{
"epoch": 1.216,
"grad_norm": 4871.5166015625,
"learning_rate": 2.976e-05,
"loss": 0.027,
"step": 760
},
{
"epoch": 1.248,
"grad_norm": 4601.7197265625,
"learning_rate": 2.922666666666667e-05,
"loss": 0.0295,
"step": 780
},
{
"epoch": 1.28,
"grad_norm": 5593.90380859375,
"learning_rate": 2.8693333333333332e-05,
"loss": 0.0271,
"step": 800
},
{
"epoch": 1.312,
"grad_norm": 6107.31298828125,
"learning_rate": 2.816e-05,
"loss": 0.0273,
"step": 820
},
{
"epoch": 1.3439999999999999,
"grad_norm": 6053.76171875,
"learning_rate": 2.7626666666666668e-05,
"loss": 0.0272,
"step": 840
}
],
"logging_steps": 20,
"max_steps": 1875,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 50,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 2.16340799422464e+16,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}