|
{ |
|
"best_metric": 0.8990825688073395, |
|
"best_model_checkpoint": "google-vit-base-patch16-224-cartoon-emotion-detection/checkpoint-56", |
|
"epoch": 9.969696969696969, |
|
"global_step": 80, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.97, |
|
"eval_accuracy": 0.5596330275229358, |
|
"eval_f1": 0.5360425549075453, |
|
"eval_loss": 0.9901540279388428, |
|
"eval_precision": 0.5505540964988554, |
|
"eval_recall": 0.5596330275229358, |
|
"eval_runtime": 5.4111, |
|
"eval_samples_per_second": 20.144, |
|
"eval_steps_per_second": 0.37, |
|
"step": 8 |
|
}, |
|
{ |
|
"epoch": 1.24, |
|
"learning_rate": 0.00011666666666666667, |
|
"loss": 1.242, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 1.97, |
|
"eval_accuracy": 0.8165137614678899, |
|
"eval_f1": 0.8131755577433994, |
|
"eval_loss": 0.5157326459884644, |
|
"eval_precision": 0.819457058574903, |
|
"eval_recall": 0.8165137614678899, |
|
"eval_runtime": 4.6435, |
|
"eval_samples_per_second": 23.473, |
|
"eval_steps_per_second": 0.431, |
|
"step": 16 |
|
}, |
|
{ |
|
"epoch": 2.48, |
|
"learning_rate": 0.0001, |
|
"loss": 0.4438, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 2.97, |
|
"eval_accuracy": 0.8440366972477065, |
|
"eval_f1": 0.8446079913969823, |
|
"eval_loss": 0.3870726525783539, |
|
"eval_precision": 0.8516382699868937, |
|
"eval_recall": 0.8440366972477065, |
|
"eval_runtime": 4.4313, |
|
"eval_samples_per_second": 24.598, |
|
"eval_steps_per_second": 0.451, |
|
"step": 24 |
|
}, |
|
{ |
|
"epoch": 3.73, |
|
"learning_rate": 8.333333333333333e-05, |
|
"loss": 0.1768, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 3.97, |
|
"eval_accuracy": 0.8623853211009175, |
|
"eval_f1": 0.8585073146382761, |
|
"eval_loss": 0.353137731552124, |
|
"eval_precision": 0.865271038673443, |
|
"eval_recall": 0.8623853211009175, |
|
"eval_runtime": 6.7356, |
|
"eval_samples_per_second": 16.183, |
|
"eval_steps_per_second": 0.297, |
|
"step": 32 |
|
}, |
|
{ |
|
"epoch": 4.97, |
|
"learning_rate": 6.666666666666667e-05, |
|
"loss": 0.0661, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 4.97, |
|
"eval_accuracy": 0.8715596330275229, |
|
"eval_f1": 0.8673695820239424, |
|
"eval_loss": 0.3780345022678375, |
|
"eval_precision": 0.8693191511585397, |
|
"eval_recall": 0.8715596330275229, |
|
"eval_runtime": 4.3244, |
|
"eval_samples_per_second": 25.206, |
|
"eval_steps_per_second": 0.462, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 5.97, |
|
"eval_accuracy": 0.8623853211009175, |
|
"eval_f1": 0.8631844663686221, |
|
"eval_loss": 0.374713659286499, |
|
"eval_precision": 0.864939209175348, |
|
"eval_recall": 0.8623853211009175, |
|
"eval_runtime": 4.479, |
|
"eval_samples_per_second": 24.336, |
|
"eval_steps_per_second": 0.447, |
|
"step": 48 |
|
}, |
|
{ |
|
"epoch": 6.24, |
|
"learning_rate": 5e-05, |
|
"loss": 0.0375, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 6.97, |
|
"eval_accuracy": 0.8990825688073395, |
|
"eval_f1": 0.8970800861485594, |
|
"eval_loss": 0.37597015500068665, |
|
"eval_precision": 0.8961258037611989, |
|
"eval_recall": 0.8990825688073395, |
|
"eval_runtime": 4.4765, |
|
"eval_samples_per_second": 24.349, |
|
"eval_steps_per_second": 0.447, |
|
"step": 56 |
|
}, |
|
{ |
|
"epoch": 7.48, |
|
"learning_rate": 3.3333333333333335e-05, |
|
"loss": 0.0362, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 7.97, |
|
"eval_accuracy": 0.8715596330275229, |
|
"eval_f1": 0.8680866720523425, |
|
"eval_loss": 0.4092080295085907, |
|
"eval_precision": 0.8684143462632116, |
|
"eval_recall": 0.8715596330275229, |
|
"eval_runtime": 4.3876, |
|
"eval_samples_per_second": 24.843, |
|
"eval_steps_per_second": 0.456, |
|
"step": 64 |
|
}, |
|
{ |
|
"epoch": 8.73, |
|
"learning_rate": 1.6666666666666667e-05, |
|
"loss": 0.0322, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 8.97, |
|
"eval_accuracy": 0.8899082568807339, |
|
"eval_f1": 0.8888099976565806, |
|
"eval_loss": 0.3499366044998169, |
|
"eval_precision": 0.8880198424442283, |
|
"eval_recall": 0.8899082568807339, |
|
"eval_runtime": 5.3381, |
|
"eval_samples_per_second": 20.419, |
|
"eval_steps_per_second": 0.375, |
|
"step": 72 |
|
}, |
|
{ |
|
"epoch": 9.97, |
|
"learning_rate": 0.0, |
|
"loss": 0.029, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 9.97, |
|
"eval_accuracy": 0.8807339449541285, |
|
"eval_f1": 0.8782945902988435, |
|
"eval_loss": 0.37059423327445984, |
|
"eval_precision": 0.8768597487153273, |
|
"eval_recall": 0.8807339449541285, |
|
"eval_runtime": 4.4763, |
|
"eval_samples_per_second": 24.35, |
|
"eval_steps_per_second": 0.447, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 9.97, |
|
"step": 80, |
|
"total_flos": 1.6024080844892897e+18, |
|
"train_loss": 0.2579587835818529, |
|
"train_runtime": 415.4218, |
|
"train_samples_per_second": 49.829, |
|
"train_steps_per_second": 0.193 |
|
} |
|
], |
|
"max_steps": 80, |
|
"num_train_epochs": 10, |
|
"total_flos": 1.6024080844892897e+18, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|