|
{ |
|
"best_metric": 0.8094336953840884, |
|
"best_model_checkpoint": "distilhubert-finetuned-mixed-data/checkpoint-45", |
|
"epoch": 10.0, |
|
"eval_steps": 500, |
|
"global_step": 45, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.8888888888888888, |
|
"eval_accuracy": 0.63003663003663, |
|
"eval_confusion_matrix": [ |
|
[ |
|
53, |
|
0, |
|
16, |
|
3 |
|
], |
|
[ |
|
28, |
|
1, |
|
31, |
|
0 |
|
], |
|
[ |
|
13, |
|
0, |
|
62, |
|
1 |
|
], |
|
[ |
|
3, |
|
0, |
|
6, |
|
56 |
|
] |
|
], |
|
"eval_f1": 0.566693372031096, |
|
"eval_loss": 1.31327486038208, |
|
"eval_precision": 0.7361933549293478, |
|
"eval_recall": 0.63003663003663, |
|
"eval_runtime": 2.8436, |
|
"eval_samples_per_second": 96.005, |
|
"eval_steps_per_second": 1.055, |
|
"step": 4 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_accuracy": 0.6593406593406593, |
|
"eval_confusion_matrix": [ |
|
[ |
|
41, |
|
10, |
|
16, |
|
5 |
|
], |
|
[ |
|
21, |
|
1, |
|
38, |
|
0 |
|
], |
|
[ |
|
1, |
|
2, |
|
73, |
|
0 |
|
], |
|
[ |
|
0, |
|
0, |
|
0, |
|
65 |
|
] |
|
], |
|
"eval_f1": 0.595713773130153, |
|
"eval_loss": 0.9242589473724365, |
|
"eval_precision": 0.5696509512811321, |
|
"eval_recall": 0.6593406593406593, |
|
"eval_runtime": 2.8668, |
|
"eval_samples_per_second": 95.227, |
|
"eval_steps_per_second": 1.046, |
|
"step": 9 |
|
}, |
|
{ |
|
"epoch": 2.888888888888889, |
|
"eval_accuracy": 0.717948717948718, |
|
"eval_confusion_matrix": [ |
|
[ |
|
52, |
|
8, |
|
7, |
|
5 |
|
], |
|
[ |
|
18, |
|
5, |
|
37, |
|
0 |
|
], |
|
[ |
|
1, |
|
0, |
|
75, |
|
0 |
|
], |
|
[ |
|
1, |
|
0, |
|
0, |
|
64 |
|
] |
|
], |
|
"eval_f1": 0.6621621567125869, |
|
"eval_loss": 0.8232662081718445, |
|
"eval_precision": 0.671303801513745, |
|
"eval_recall": 0.717948717948718, |
|
"eval_runtime": 2.8711, |
|
"eval_samples_per_second": 95.084, |
|
"eval_steps_per_second": 1.045, |
|
"step": 13 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"eval_accuracy": 0.6959706959706959, |
|
"eval_confusion_matrix": [ |
|
[ |
|
45, |
|
9, |
|
13, |
|
5 |
|
], |
|
[ |
|
13, |
|
5, |
|
42, |
|
0 |
|
], |
|
[ |
|
1, |
|
0, |
|
75, |
|
0 |
|
], |
|
[ |
|
0, |
|
0, |
|
0, |
|
65 |
|
] |
|
], |
|
"eval_f1": 0.6428796900399459, |
|
"eval_loss": 0.951453447341919, |
|
"eval_precision": 0.6613447705829899, |
|
"eval_recall": 0.6959706959706959, |
|
"eval_runtime": 2.8813, |
|
"eval_samples_per_second": 94.75, |
|
"eval_steps_per_second": 1.041, |
|
"step": 18 |
|
}, |
|
{ |
|
"epoch": 4.888888888888889, |
|
"eval_accuracy": 0.6666666666666666, |
|
"eval_confusion_matrix": [ |
|
[ |
|
16, |
|
49, |
|
3, |
|
4 |
|
], |
|
[ |
|
1, |
|
26, |
|
33, |
|
0 |
|
], |
|
[ |
|
0, |
|
1, |
|
75, |
|
0 |
|
], |
|
[ |
|
0, |
|
0, |
|
0, |
|
65 |
|
] |
|
], |
|
"eval_f1": 0.6331541584523206, |
|
"eval_loss": 1.0080645084381104, |
|
"eval_precision": 0.7358031394913882, |
|
"eval_recall": 0.6666666666666666, |
|
"eval_runtime": 2.9108, |
|
"eval_samples_per_second": 93.789, |
|
"eval_steps_per_second": 1.031, |
|
"step": 22 |
|
}, |
|
{ |
|
"epoch": 6.0, |
|
"eval_accuracy": 0.652014652014652, |
|
"eval_confusion_matrix": [ |
|
[ |
|
66, |
|
0, |
|
0, |
|
6 |
|
], |
|
[ |
|
36, |
|
16, |
|
8, |
|
0 |
|
], |
|
[ |
|
10, |
|
35, |
|
31, |
|
0 |
|
], |
|
[ |
|
0, |
|
0, |
|
0, |
|
65 |
|
] |
|
], |
|
"eval_f1": 0.6302407955860642, |
|
"eval_loss": 0.9598046541213989, |
|
"eval_precision": 0.6636241740077424, |
|
"eval_recall": 0.652014652014652, |
|
"eval_runtime": 2.9773, |
|
"eval_samples_per_second": 91.693, |
|
"eval_steps_per_second": 1.008, |
|
"step": 27 |
|
}, |
|
{ |
|
"epoch": 6.888888888888889, |
|
"eval_accuracy": 0.7692307692307693, |
|
"eval_confusion_matrix": [ |
|
[ |
|
66, |
|
0, |
|
0, |
|
6 |
|
], |
|
[ |
|
29, |
|
21, |
|
10, |
|
0 |
|
], |
|
[ |
|
3, |
|
15, |
|
58, |
|
0 |
|
], |
|
[ |
|
0, |
|
0, |
|
0, |
|
65 |
|
] |
|
], |
|
"eval_f1": 0.7527855586679116, |
|
"eval_loss": 0.7898163795471191, |
|
"eval_precision": 0.7612467677056892, |
|
"eval_recall": 0.7692307692307693, |
|
"eval_runtime": 2.8918, |
|
"eval_samples_per_second": 94.404, |
|
"eval_steps_per_second": 1.037, |
|
"step": 31 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"eval_accuracy": 0.7802197802197802, |
|
"eval_confusion_matrix": [ |
|
[ |
|
66, |
|
0, |
|
0, |
|
6 |
|
], |
|
[ |
|
25, |
|
15, |
|
19, |
|
1 |
|
], |
|
[ |
|
1, |
|
6, |
|
68, |
|
1 |
|
], |
|
[ |
|
1, |
|
0, |
|
0, |
|
64 |
|
] |
|
], |
|
"eval_f1": 0.7471179200524057, |
|
"eval_loss": 0.7337484359741211, |
|
"eval_precision": 0.7733845922309294, |
|
"eval_recall": 0.7802197802197802, |
|
"eval_runtime": 2.9006, |
|
"eval_samples_per_second": 94.119, |
|
"eval_steps_per_second": 1.034, |
|
"step": 36 |
|
}, |
|
{ |
|
"epoch": 8.88888888888889, |
|
"eval_accuracy": 0.7912087912087912, |
|
"eval_confusion_matrix": [ |
|
[ |
|
57, |
|
8, |
|
3, |
|
4 |
|
], |
|
[ |
|
13, |
|
24, |
|
23, |
|
0 |
|
], |
|
[ |
|
0, |
|
5, |
|
71, |
|
0 |
|
], |
|
[ |
|
1, |
|
0, |
|
0, |
|
64 |
|
] |
|
], |
|
"eval_f1": 0.7766564722186922, |
|
"eval_loss": 0.7148727178573608, |
|
"eval_precision": 0.7821504483074875, |
|
"eval_recall": 0.7912087912087912, |
|
"eval_runtime": 2.8866, |
|
"eval_samples_per_second": 94.576, |
|
"eval_steps_per_second": 1.039, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"eval_accuracy": 0.8168498168498168, |
|
"eval_confusion_matrix": [ |
|
[ |
|
51, |
|
8, |
|
1, |
|
12 |
|
], |
|
[ |
|
10, |
|
36, |
|
14, |
|
0 |
|
], |
|
[ |
|
0, |
|
5, |
|
71, |
|
0 |
|
], |
|
[ |
|
0, |
|
0, |
|
0, |
|
65 |
|
] |
|
], |
|
"eval_f1": 0.8094336953840884, |
|
"eval_loss": 0.7574812173843384, |
|
"eval_precision": 0.8127936625684181, |
|
"eval_recall": 0.8168498168498168, |
|
"eval_runtime": 2.8967, |
|
"eval_samples_per_second": 94.245, |
|
"eval_steps_per_second": 1.036, |
|
"step": 45 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 180, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 45, |
|
"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": 2.4789069504e+16, |
|
"train_batch_size": 128, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|