|
{ |
|
"best_metric": 0.8860103626943006, |
|
"best_model_checkpoint": "./vit-focal-skin/checkpoint-1252", |
|
"epoch": 4.0, |
|
"global_step": 2504, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 0.00019201277955271565, |
|
"loss": 0.7905, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 0.00018402555910543132, |
|
"loss": 0.6162, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 0.000176038338658147, |
|
"loss": 0.5709, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 0.00016805111821086263, |
|
"loss": 0.5101, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 0.0001600638977635783, |
|
"loss": 0.4966, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 0.00015207667731629394, |
|
"loss": 0.4563, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_accuracy": 0.8341968911917098, |
|
"eval_f1": 0.8331282144797781, |
|
"eval_loss": 0.40268635749816895, |
|
"eval_precision": 0.8791963514680484, |
|
"eval_recall": 0.8341968911917098, |
|
"eval_runtime": 1.5386, |
|
"eval_samples_per_second": 125.44, |
|
"eval_steps_per_second": 16.249, |
|
"step": 626 |
|
}, |
|
{ |
|
"epoch": 1.12, |
|
"learning_rate": 0.00014408945686900958, |
|
"loss": 0.385, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 1.28, |
|
"learning_rate": 0.00013610223642172525, |
|
"loss": 0.3621, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 1.44, |
|
"learning_rate": 0.00012811501597444092, |
|
"loss": 0.3624, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 1.6, |
|
"learning_rate": 0.00012012779552715656, |
|
"loss": 0.3403, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.76, |
|
"learning_rate": 0.00011214057507987221, |
|
"loss": 0.3293, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 1.92, |
|
"learning_rate": 0.00010415335463258787, |
|
"loss": 0.2954, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_accuracy": 0.8860103626943006, |
|
"eval_f1": 0.8891954516071332, |
|
"eval_loss": 0.3063604235649109, |
|
"eval_precision": 0.8988400162775354, |
|
"eval_recall": 0.8860103626943006, |
|
"eval_runtime": 1.5206, |
|
"eval_samples_per_second": 126.921, |
|
"eval_steps_per_second": 16.441, |
|
"step": 1252 |
|
}, |
|
{ |
|
"epoch": 2.08, |
|
"learning_rate": 9.616613418530351e-05, |
|
"loss": 0.2258, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 2.24, |
|
"learning_rate": 8.817891373801918e-05, |
|
"loss": 0.1583, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 2.4, |
|
"learning_rate": 8.019169329073483e-05, |
|
"loss": 0.1589, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 2.56, |
|
"learning_rate": 7.220447284345049e-05, |
|
"loss": 0.1271, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 2.72, |
|
"learning_rate": 6.421725239616614e-05, |
|
"loss": 0.1746, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 2.88, |
|
"learning_rate": 5.623003194888179e-05, |
|
"loss": 0.1118, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_accuracy": 0.8704663212435233, |
|
"eval_f1": 0.8663258953141536, |
|
"eval_loss": 0.30049628019332886, |
|
"eval_precision": 0.8698602729520133, |
|
"eval_recall": 0.8704663212435233, |
|
"eval_runtime": 1.4368, |
|
"eval_samples_per_second": 134.33, |
|
"eval_steps_per_second": 17.4, |
|
"step": 1878 |
|
}, |
|
{ |
|
"epoch": 3.04, |
|
"learning_rate": 4.824281150159744e-05, |
|
"loss": 0.1193, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 3.19, |
|
"learning_rate": 4.0255591054313104e-05, |
|
"loss": 0.0409, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 3.35, |
|
"learning_rate": 3.226837060702875e-05, |
|
"loss": 0.0327, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 3.51, |
|
"learning_rate": 2.428115015974441e-05, |
|
"loss": 0.0357, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 3.67, |
|
"learning_rate": 1.6293929712460065e-05, |
|
"loss": 0.0386, |
|
"step": 2300 |
|
}, |
|
{ |
|
"epoch": 3.83, |
|
"learning_rate": 8.306709265175718e-06, |
|
"loss": 0.0303, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 3.99, |
|
"learning_rate": 3.194888178913738e-07, |
|
"loss": 0.0317, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"eval_accuracy": 0.8549222797927462, |
|
"eval_f1": 0.8560478324319817, |
|
"eval_loss": 0.3552953004837036, |
|
"eval_precision": 0.8595484172497833, |
|
"eval_recall": 0.8549222797927462, |
|
"eval_runtime": 1.5113, |
|
"eval_samples_per_second": 127.703, |
|
"eval_steps_per_second": 16.542, |
|
"step": 2504 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"step": 2504, |
|
"total_flos": 3.104468219559813e+18, |
|
"train_loss": 0.2716429328141264, |
|
"train_runtime": 832.7663, |
|
"train_samples_per_second": 48.105, |
|
"train_steps_per_second": 3.007 |
|
} |
|
], |
|
"max_steps": 2504, |
|
"num_train_epochs": 4, |
|
"total_flos": 3.104468219559813e+18, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|