Devarshi's picture
End of training
bdbd32e
{
"best_metric": 0.9960906958561376,
"best_model_checkpoint": "Brain_Tumor_Classification_using_swin/checkpoint-540",
"epoch": 3.0,
"global_step": 540,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.06,
"learning_rate": 9.259259259259259e-06,
"loss": 1.0953,
"step": 10
},
{
"epoch": 0.11,
"learning_rate": 1.8518518518518518e-05,
"loss": 0.9167,
"step": 20
},
{
"epoch": 0.17,
"learning_rate": 2.777777777777778e-05,
"loss": 0.5877,
"step": 30
},
{
"epoch": 0.22,
"learning_rate": 3.7037037037037037e-05,
"loss": 0.3904,
"step": 40
},
{
"epoch": 0.28,
"learning_rate": 4.62962962962963e-05,
"loss": 0.3268,
"step": 50
},
{
"epoch": 0.33,
"learning_rate": 4.938271604938271e-05,
"loss": 0.3087,
"step": 60
},
{
"epoch": 0.39,
"learning_rate": 4.835390946502058e-05,
"loss": 0.2053,
"step": 70
},
{
"epoch": 0.44,
"learning_rate": 4.732510288065844e-05,
"loss": 0.1831,
"step": 80
},
{
"epoch": 0.5,
"learning_rate": 4.62962962962963e-05,
"loss": 0.1992,
"step": 90
},
{
"epoch": 0.56,
"learning_rate": 4.5267489711934157e-05,
"loss": 0.2218,
"step": 100
},
{
"epoch": 0.61,
"learning_rate": 4.423868312757202e-05,
"loss": 0.2151,
"step": 110
},
{
"epoch": 0.67,
"learning_rate": 4.3209876543209875e-05,
"loss": 0.1468,
"step": 120
},
{
"epoch": 0.72,
"learning_rate": 4.2181069958847744e-05,
"loss": 0.1526,
"step": 130
},
{
"epoch": 0.78,
"learning_rate": 4.11522633744856e-05,
"loss": 0.1269,
"step": 140
},
{
"epoch": 0.83,
"learning_rate": 4.012345679012346e-05,
"loss": 0.1384,
"step": 150
},
{
"epoch": 0.89,
"learning_rate": 3.909465020576132e-05,
"loss": 0.127,
"step": 160
},
{
"epoch": 0.94,
"learning_rate": 3.806584362139918e-05,
"loss": 0.1184,
"step": 170
},
{
"epoch": 1.0,
"learning_rate": 3.7037037037037037e-05,
"loss": 0.1234,
"step": 180
},
{
"epoch": 1.0,
"eval_accuracy": 0.9839718530101642,
"eval_f1": 0.9839718530101642,
"eval_loss": 0.045045629143714905,
"eval_precision": 0.9839718530101642,
"eval_recall": 0.9839718530101642,
"eval_runtime": 439.4532,
"eval_samples_per_second": 5.821,
"eval_steps_per_second": 0.182,
"step": 180
},
{
"epoch": 1.06,
"learning_rate": 3.60082304526749e-05,
"loss": 0.1162,
"step": 190
},
{
"epoch": 1.11,
"learning_rate": 3.497942386831276e-05,
"loss": 0.1137,
"step": 200
},
{
"epoch": 1.17,
"learning_rate": 3.395061728395062e-05,
"loss": 0.1066,
"step": 210
},
{
"epoch": 1.22,
"learning_rate": 3.292181069958848e-05,
"loss": 0.0998,
"step": 220
},
{
"epoch": 1.28,
"learning_rate": 3.1893004115226336e-05,
"loss": 0.0899,
"step": 230
},
{
"epoch": 1.33,
"learning_rate": 3.08641975308642e-05,
"loss": 0.0929,
"step": 240
},
{
"epoch": 1.39,
"learning_rate": 2.9835390946502057e-05,
"loss": 0.1265,
"step": 250
},
{
"epoch": 1.44,
"learning_rate": 2.880658436213992e-05,
"loss": 0.1177,
"step": 260
},
{
"epoch": 1.5,
"learning_rate": 2.777777777777778e-05,
"loss": 0.0866,
"step": 270
},
{
"epoch": 1.56,
"learning_rate": 2.6748971193415638e-05,
"loss": 0.1129,
"step": 280
},
{
"epoch": 1.61,
"learning_rate": 2.5720164609053497e-05,
"loss": 0.087,
"step": 290
},
{
"epoch": 1.67,
"learning_rate": 2.4691358024691357e-05,
"loss": 0.103,
"step": 300
},
{
"epoch": 1.72,
"learning_rate": 2.366255144032922e-05,
"loss": 0.0784,
"step": 310
},
{
"epoch": 1.78,
"learning_rate": 2.2633744855967078e-05,
"loss": 0.0833,
"step": 320
},
{
"epoch": 1.83,
"learning_rate": 2.1604938271604937e-05,
"loss": 0.0847,
"step": 330
},
{
"epoch": 1.89,
"learning_rate": 2.05761316872428e-05,
"loss": 0.0738,
"step": 340
},
{
"epoch": 1.94,
"learning_rate": 1.954732510288066e-05,
"loss": 0.0844,
"step": 350
},
{
"epoch": 2.0,
"learning_rate": 1.8518518518518518e-05,
"loss": 0.0837,
"step": 360
},
{
"epoch": 2.0,
"eval_accuracy": 0.9925723221266615,
"eval_f1": 0.9925723221266615,
"eval_loss": 0.019814323633909225,
"eval_precision": 0.9925723221266615,
"eval_recall": 0.9925723221266615,
"eval_runtime": 440.2071,
"eval_samples_per_second": 5.811,
"eval_steps_per_second": 0.182,
"step": 360
},
{
"epoch": 2.06,
"learning_rate": 1.748971193415638e-05,
"loss": 0.0742,
"step": 370
},
{
"epoch": 2.11,
"learning_rate": 1.646090534979424e-05,
"loss": 0.0805,
"step": 380
},
{
"epoch": 2.17,
"learning_rate": 1.54320987654321e-05,
"loss": 0.0892,
"step": 390
},
{
"epoch": 2.22,
"learning_rate": 1.440329218106996e-05,
"loss": 0.0732,
"step": 400
},
{
"epoch": 2.28,
"learning_rate": 1.3374485596707819e-05,
"loss": 0.0823,
"step": 410
},
{
"epoch": 2.33,
"learning_rate": 1.2345679012345678e-05,
"loss": 0.0623,
"step": 420
},
{
"epoch": 2.39,
"learning_rate": 1.1316872427983539e-05,
"loss": 0.0609,
"step": 430
},
{
"epoch": 2.44,
"learning_rate": 1.02880658436214e-05,
"loss": 0.0752,
"step": 440
},
{
"epoch": 2.5,
"learning_rate": 9.259259259259259e-06,
"loss": 0.0608,
"step": 450
},
{
"epoch": 2.56,
"learning_rate": 8.23045267489712e-06,
"loss": 0.0529,
"step": 460
},
{
"epoch": 2.61,
"learning_rate": 7.20164609053498e-06,
"loss": 0.0628,
"step": 470
},
{
"epoch": 2.67,
"learning_rate": 6.172839506172839e-06,
"loss": 0.0508,
"step": 480
},
{
"epoch": 2.72,
"learning_rate": 5.1440329218107e-06,
"loss": 0.0517,
"step": 490
},
{
"epoch": 2.78,
"learning_rate": 4.11522633744856e-06,
"loss": 0.06,
"step": 500
},
{
"epoch": 2.83,
"learning_rate": 3.0864197530864196e-06,
"loss": 0.0671,
"step": 510
},
{
"epoch": 2.89,
"learning_rate": 2.05761316872428e-06,
"loss": 0.0733,
"step": 520
},
{
"epoch": 2.94,
"learning_rate": 1.02880658436214e-06,
"loss": 0.0558,
"step": 530
},
{
"epoch": 3.0,
"learning_rate": 0.0,
"loss": 0.0373,
"step": 540
},
{
"epoch": 3.0,
"eval_accuracy": 0.9960906958561376,
"eval_f1": 0.9960906958561376,
"eval_loss": 0.012254622764885426,
"eval_precision": 0.9960906958561376,
"eval_recall": 0.9960906958561376,
"eval_runtime": 442.7035,
"eval_samples_per_second": 5.778,
"eval_steps_per_second": 0.181,
"step": 540
},
{
"epoch": 3.0,
"step": 540,
"total_flos": 5.409667845997019e+18,
"train_loss": 0.15731672859854168,
"train_runtime": 52202.335,
"train_samples_per_second": 1.323,
"train_steps_per_second": 0.01
}
],
"max_steps": 540,
"num_train_epochs": 3,
"total_flos": 5.409667845997019e+18,
"trial_name": null,
"trial_params": null
}