teapotllm / checkpoint-219 /trainer_state.json
zakerytclarke's picture
Upload folder using huggingface_hub
5aa6f84 verified
raw
history blame contribute delete
2.11 kB
{
"best_metric": 0.04732182249426842,
"best_model_checkpoint": "./teapotllm/checkpoint-219",
"epoch": 1.0,
"eval_steps": 500,
"global_step": 219,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"em_boolean": 0.5294117647058824,
"em_chat": 0.03783783783783784,
"em_extraction": 0.5333333333333333,
"em_qa": 0.5333333333333333,
"em_summarization": 0.0,
"em_unanswerable": 0.0,
"epoch": 1.0,
"mean_em": 0.12075471698113208,
"mean_similarity": 0.6624939311014594,
"mean_word_count_diff": 11.958490566037735,
"similarity_boolean": 0.649950959993636,
"similarity_chat": 0.6068178282194846,
"similarity_extraction": 0.8274670541286469,
"similarity_qa": 0.8895131280024846,
"similarity_summarization": 0.7897715005609725,
"similarity_unanswerable": 0.8186558306217193,
"word_count_diff_boolean": 2.764705882352941,
"word_count_diff_chat": 11.275675675675675,
"word_count_diff_extraction": 1.8666666666666667,
"word_count_diff_qa": 2.7333333333333334,
"word_count_diff_summarization": 50.05555555555556,
"word_count_diff_unanswerable": 4.4
},
{
"epoch": 1.0,
"grad_norm": 0.29226627945899963,
"learning_rate": 1.314e-05,
"loss": 0.0548,
"step": 219
},
{
"epoch": 1.0,
"eval_loss": 0.04732182249426842,
"eval_runtime": 17.2425,
"eval_samples_per_second": 15.369,
"eval_steps_per_second": 1.972,
"step": 219
}
],
"logging_steps": 500,
"max_steps": 2190,
"num_input_tokens_seen": 0,
"num_train_epochs": 10,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 2012062421090304.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}