| { | |
| "config": { | |
| "model_name": "bert-base-uncased", | |
| "seed": 42, | |
| "normalize_labels": true, | |
| "nouns_only": true, | |
| "clean_words_only": true, | |
| "epochs": 8.0, | |
| "learning_rate": 2e-05, | |
| "train_batch_size": 64, | |
| "eval_batch_size": 128, | |
| "grad_accumulation": 1, | |
| "eval_size": 0.1 | |
| }, | |
| "data_size_total": 4196, | |
| "data_size_train": 3776, | |
| "data_size_eval": 420, | |
| "final_model_dir": "outputs/mrc_bert_full_eval10/final_model", | |
| "eval_metrics": { | |
| "eval_loss": 0.008318813517689705, | |
| "eval_mae": 40.69350791828973, | |
| "eval_rmse": 54.724519092881394, | |
| "eval_pearson": 0.8696628446003037, | |
| "eval_runtime": 0.4034, | |
| "eval_samples_per_second": 1041.075, | |
| "eval_steps_per_second": 9.915, | |
| "epoch": 8.0 | |
| } | |
| } |