djbp's picture
End of training
f8ab760 verified
raw
history blame
6.27 kB
{
"best_metric": 0.8333333333333334,
"best_model_checkpoint": "swin-base-patch4-window7-224-in22k-cons_Classification_base_V10/checkpoint-2",
"epoch": 7.0,
"eval_steps": 500,
"global_step": 7,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"step": 1,
"train_accuracy": 0.26666666666666666,
"train_auc_class_0": 0.32000000000000006,
"train_auc_class_1": 0.6200000000000001,
"train_auc_class_2": 0.39999999999999997,
"train_auc_overall": 0.4466666666666667,
"train_loss": 1.1479226350784302,
"train_runtime": 1.6122,
"train_samples_per_second": 9.304,
"train_steps_per_second": 0.62
},
{
"epoch": 1.0,
"eval_accuracy": 0.5,
"eval_auc_class_0": 0.875,
"eval_auc_class_1": 0.5,
"eval_auc_class_2": 0.625,
"eval_auc_overall": 0.6666666666666666,
"eval_loss": 0.9469685554504395,
"eval_runtime": 0.6769,
"eval_samples_per_second": 8.863,
"eval_steps_per_second": 1.477,
"step": 1
},
{
"epoch": 2.0,
"step": 2,
"train_accuracy": 0.6,
"train_auc_class_0": 0.82,
"train_auc_class_1": 0.96,
"train_auc_class_2": 0.9,
"train_auc_overall": 0.8933333333333332,
"train_loss": 0.814429521560669,
"train_runtime": 1.6517,
"train_samples_per_second": 9.081,
"train_steps_per_second": 0.605
},
{
"epoch": 2.0,
"eval_accuracy": 0.8333333333333334,
"eval_auc_class_0": 1.0,
"eval_auc_class_1": 1.0,
"eval_auc_class_2": 1.0,
"eval_auc_overall": 1.0,
"eval_loss": 0.8139967918395996,
"eval_runtime": 0.7014,
"eval_samples_per_second": 8.555,
"eval_steps_per_second": 1.426,
"step": 2
},
{
"epoch": 3.0,
"step": 3,
"train_accuracy": 0.8666666666666667,
"train_auc_class_0": 1.0,
"train_auc_class_1": 1.0,
"train_auc_class_2": 0.9600000000000001,
"train_auc_overall": 0.9866666666666667,
"train_loss": 0.6608889698982239,
"train_runtime": 1.7063,
"train_samples_per_second": 8.791,
"train_steps_per_second": 0.586
},
{
"epoch": 3.0,
"eval_accuracy": 0.8333333333333334,
"eval_auc_class_0": 1.0,
"eval_auc_class_1": 1.0,
"eval_auc_class_2": 1.0,
"eval_auc_overall": 1.0,
"eval_loss": 0.7299937605857849,
"eval_runtime": 0.7538,
"eval_samples_per_second": 7.959,
"eval_steps_per_second": 1.327,
"step": 3
},
{
"epoch": 4.0,
"step": 4,
"train_accuracy": 1.0,
"train_auc_class_0": 1.0,
"train_auc_class_1": 1.0,
"train_auc_class_2": 1.0,
"train_auc_overall": 1.0,
"train_loss": 0.42420145869255066,
"train_runtime": 1.6223,
"train_samples_per_second": 9.246,
"train_steps_per_second": 0.616
},
{
"epoch": 4.0,
"eval_accuracy": 0.8333333333333334,
"eval_auc_class_0": 1.0,
"eval_auc_class_1": 1.0,
"eval_auc_class_2": 1.0,
"eval_auc_overall": 1.0,
"eval_loss": 0.6729159355163574,
"eval_runtime": 0.772,
"eval_samples_per_second": 7.772,
"eval_steps_per_second": 1.295,
"step": 4
},
{
"epoch": 5.0,
"step": 5,
"train_accuracy": 0.9333333333333333,
"train_auc_class_0": 1.0,
"train_auc_class_1": 1.0,
"train_auc_class_2": 1.0,
"train_auc_overall": 1.0,
"train_loss": 0.42396074533462524,
"train_runtime": 1.6518,
"train_samples_per_second": 9.081,
"train_steps_per_second": 0.605
},
{
"epoch": 5.0,
"eval_accuracy": 0.8333333333333334,
"eval_auc_class_0": 1.0,
"eval_auc_class_1": 1.0,
"eval_auc_class_2": 1.0,
"eval_auc_overall": 1.0,
"eval_loss": 0.6391382813453674,
"eval_runtime": 0.6917,
"eval_samples_per_second": 8.675,
"eval_steps_per_second": 1.446,
"step": 5
},
{
"epoch": 6.0,
"step": 6,
"train_accuracy": 1.0,
"train_auc_class_0": 1.0,
"train_auc_class_1": 1.0,
"train_auc_class_2": 1.0,
"train_auc_overall": 1.0,
"train_loss": 0.29362058639526367,
"train_runtime": 1.6474,
"train_samples_per_second": 9.105,
"train_steps_per_second": 0.607
},
{
"epoch": 6.0,
"eval_accuracy": 0.8333333333333334,
"eval_auc_class_0": 1.0,
"eval_auc_class_1": 1.0,
"eval_auc_class_2": 1.0,
"eval_auc_overall": 1.0,
"eval_loss": 0.6181725859642029,
"eval_runtime": 0.6858,
"eval_samples_per_second": 8.748,
"eval_steps_per_second": 1.458,
"step": 6
},
{
"epoch": 7.0,
"step": 7,
"train_accuracy": 1.0,
"train_auc_class_0": 1.0,
"train_auc_class_1": 1.0,
"train_auc_class_2": 1.0,
"train_auc_overall": 1.0,
"train_loss": 0.35040372610092163,
"train_runtime": 1.7724,
"train_samples_per_second": 8.463,
"train_steps_per_second": 0.564
},
{
"epoch": 7.0,
"eval_accuracy": 0.8333333333333334,
"eval_auc_class_0": 1.0,
"eval_auc_class_1": 1.0,
"eval_auc_class_2": 1.0,
"eval_auc_overall": 1.0,
"eval_loss": 0.6054350733757019,
"eval_runtime": 0.7011,
"eval_samples_per_second": 8.558,
"eval_steps_per_second": 1.426,
"step": 7
},
{
"epoch": 7.0,
"step": 7,
"total_flos": 8226380544399360.0,
"train_loss": 0.17954584530421666,
"train_runtime": 78.6151,
"train_samples_per_second": 1.336,
"train_steps_per_second": 0.089
}
],
"logging_steps": 10,
"max_steps": 7,
"num_input_tokens_seen": 0,
"num_train_epochs": 7,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 8226380544399360.0,
"train_batch_size": 128,
"trial_name": null,
"trial_params": null
}