Llama-1B-LoRA-r32 / checkpoint-200 /trainer_state.json
vijay-ravichander's picture
Training in progress, step 200, checkpoint
15b48ee verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.24390243902439024,
"eval_steps": 500,
"global_step": 200,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.024390243902439025,
"grad_norm": 0.2627893388271332,
"learning_rate": 0.00019631901840490797,
"loss": 2.6021,
"step": 20
},
{
"epoch": 0.04878048780487805,
"grad_norm": 0.22776605188846588,
"learning_rate": 0.0001914110429447853,
"loss": 2.3247,
"step": 40
},
{
"epoch": 0.07317073170731707,
"grad_norm": 0.219473734498024,
"learning_rate": 0.00018650306748466258,
"loss": 2.327,
"step": 60
},
{
"epoch": 0.0975609756097561,
"grad_norm": 0.2347700446844101,
"learning_rate": 0.00018159509202453987,
"loss": 2.2708,
"step": 80
},
{
"epoch": 0.12195121951219512,
"grad_norm": 0.22764651477336884,
"learning_rate": 0.0001766871165644172,
"loss": 2.2459,
"step": 100
},
{
"epoch": 0.14634146341463414,
"grad_norm": 0.2603573799133301,
"learning_rate": 0.0001717791411042945,
"loss": 2.1985,
"step": 120
},
{
"epoch": 0.17073170731707318,
"grad_norm": 0.30507025122642517,
"learning_rate": 0.00016687116564417177,
"loss": 2.297,
"step": 140
},
{
"epoch": 0.1951219512195122,
"grad_norm": 0.21570216119289398,
"learning_rate": 0.00016196319018404909,
"loss": 2.188,
"step": 160
},
{
"epoch": 0.21951219512195122,
"grad_norm": 0.3292004466056824,
"learning_rate": 0.0001570552147239264,
"loss": 2.1902,
"step": 180
},
{
"epoch": 0.24390243902439024,
"grad_norm": 0.24518953263759613,
"learning_rate": 0.0001521472392638037,
"loss": 2.2467,
"step": 200
}
],
"logging_steps": 20,
"max_steps": 820,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 200,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 4.87107721691136e+16,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}