Llama-3.1-8B-BookAdventures / trainer_state.json
g4rg's picture
Upload folder using huggingface_hub
99d2a01 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9795918367346939,
"eval_steps": 500,
"global_step": 12,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.08163265306122448,
"grad_norm": 10.881313562892391,
"learning_rate": 1.965925826289068e-06,
"loss": 2.0224,
"num_input_tokens_seen": 1048576,
"step": 1
},
{
"epoch": 0.16326530612244897,
"grad_norm": 13.38603350535489,
"learning_rate": 1.8660254037844386e-06,
"loss": 1.9861,
"num_input_tokens_seen": 2097152,
"step": 2
},
{
"epoch": 0.24489795918367346,
"grad_norm": 8.222813240110005,
"learning_rate": 1.7071067811865474e-06,
"loss": 1.9837,
"num_input_tokens_seen": 3145728,
"step": 3
},
{
"epoch": 0.32653061224489793,
"grad_norm": 10.396958079682259,
"learning_rate": 1.5e-06,
"loss": 1.9542,
"num_input_tokens_seen": 4194304,
"step": 4
},
{
"epoch": 0.40816326530612246,
"grad_norm": 8.858340282561143,
"learning_rate": 1.2588190451025207e-06,
"loss": 2.0559,
"num_input_tokens_seen": 5242880,
"step": 5
},
{
"epoch": 0.4897959183673469,
"grad_norm": 3.3559993754262956,
"learning_rate": 1e-06,
"loss": 1.8525,
"num_input_tokens_seen": 6291456,
"step": 6
},
{
"epoch": 0.5714285714285714,
"grad_norm": 3.1389545351229113,
"learning_rate": 7.411809548974791e-07,
"loss": 1.8691,
"num_input_tokens_seen": 7340032,
"step": 7
},
{
"epoch": 0.6530612244897959,
"grad_norm": 2.2619715751113194,
"learning_rate": 5.000000000000002e-07,
"loss": 1.8626,
"num_input_tokens_seen": 8388608,
"step": 8
},
{
"epoch": 0.7346938775510204,
"grad_norm": 2.4738597876739514,
"learning_rate": 2.9289321881345254e-07,
"loss": 1.9789,
"num_input_tokens_seen": 9437184,
"step": 9
},
{
"epoch": 0.8163265306122449,
"grad_norm": 2.169239333615782,
"learning_rate": 1.3397459621556128e-07,
"loss": 1.8399,
"num_input_tokens_seen": 10485760,
"step": 10
},
{
"epoch": 0.8979591836734694,
"grad_norm": 2.064624273077457,
"learning_rate": 3.4074173710931796e-08,
"loss": 1.8986,
"num_input_tokens_seen": 11534336,
"step": 11
},
{
"epoch": 0.9795918367346939,
"grad_norm": 2.0479265547550938,
"learning_rate": 0.0,
"loss": 1.8828,
"num_input_tokens_seen": 12582912,
"step": 12
},
{
"epoch": 0.9795918367346939,
"num_input_tokens_seen": 12582912,
"step": 12,
"total_flos": 19681687633920.0,
"train_loss": 1.9322307407855988,
"train_runtime": 728.4633,
"train_samples_per_second": 0.535,
"train_steps_per_second": 0.016
}
],
"logging_steps": 1,
"max_steps": 12,
"num_input_tokens_seen": 12582912,
"num_train_epochs": 1,
"save_steps": 25,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 19681687633920.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}