|
{ |
|
"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 |
|
} |
|
|