Brain_Tumor_Detector_swin / trainer_state.json
Devarshi's picture
End of training
b1b1461
{
"best_metric": 0.9990069513406157,
"best_model_checkpoint": "Brain_Tumor_Detector_swin/checkpoint-339",
"epoch": 3.0,
"global_step": 339,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.09,
"learning_rate": 1.4705882352941177e-05,
"loss": 0.6023,
"step": 10
},
{
"epoch": 0.18,
"learning_rate": 2.9411764705882354e-05,
"loss": 0.368,
"step": 20
},
{
"epoch": 0.27,
"learning_rate": 4.411764705882353e-05,
"loss": 0.2555,
"step": 30
},
{
"epoch": 0.35,
"learning_rate": 4.9016393442622957e-05,
"loss": 0.1834,
"step": 40
},
{
"epoch": 0.44,
"learning_rate": 4.737704918032787e-05,
"loss": 0.1955,
"step": 50
},
{
"epoch": 0.53,
"learning_rate": 4.5737704918032786e-05,
"loss": 0.168,
"step": 60
},
{
"epoch": 0.62,
"learning_rate": 4.409836065573771e-05,
"loss": 0.1057,
"step": 70
},
{
"epoch": 0.71,
"learning_rate": 4.245901639344262e-05,
"loss": 0.1323,
"step": 80
},
{
"epoch": 0.8,
"learning_rate": 4.081967213114754e-05,
"loss": 0.0936,
"step": 90
},
{
"epoch": 0.88,
"learning_rate": 3.9180327868852464e-05,
"loss": 0.0855,
"step": 100
},
{
"epoch": 0.97,
"learning_rate": 3.754098360655738e-05,
"loss": 0.079,
"step": 110
},
{
"epoch": 1.0,
"eval_accuracy": 0.9881619937694704,
"eval_f1": 0.9905894006934126,
"eval_loss": 0.02834915742278099,
"eval_precision": 0.9881422924901185,
"eval_recall": 0.9930486593843099,
"eval_runtime": 203.14,
"eval_samples_per_second": 7.901,
"eval_steps_per_second": 0.251,
"step": 113
},
{
"epoch": 1.06,
"learning_rate": 3.590163934426229e-05,
"loss": 0.0843,
"step": 120
},
{
"epoch": 1.15,
"learning_rate": 3.4262295081967214e-05,
"loss": 0.0651,
"step": 130
},
{
"epoch": 1.24,
"learning_rate": 3.2622950819672136e-05,
"loss": 0.0893,
"step": 140
},
{
"epoch": 1.33,
"learning_rate": 3.098360655737705e-05,
"loss": 0.093,
"step": 150
},
{
"epoch": 1.42,
"learning_rate": 2.934426229508197e-05,
"loss": 0.0666,
"step": 160
},
{
"epoch": 1.5,
"learning_rate": 2.7704918032786886e-05,
"loss": 0.0601,
"step": 170
},
{
"epoch": 1.59,
"learning_rate": 2.6065573770491804e-05,
"loss": 0.0734,
"step": 180
},
{
"epoch": 1.68,
"learning_rate": 2.442622950819672e-05,
"loss": 0.0523,
"step": 190
},
{
"epoch": 1.77,
"learning_rate": 2.278688524590164e-05,
"loss": 0.0572,
"step": 200
},
{
"epoch": 1.86,
"learning_rate": 2.114754098360656e-05,
"loss": 0.0415,
"step": 210
},
{
"epoch": 1.95,
"learning_rate": 1.9508196721311475e-05,
"loss": 0.0575,
"step": 220
},
{
"epoch": 2.0,
"eval_accuracy": 0.9956386292834891,
"eval_f1": 0.9965191447041274,
"eval_loss": 0.01207876205444336,
"eval_precision": 0.99800796812749,
"eval_recall": 0.9950347567030785,
"eval_runtime": 193.9311,
"eval_samples_per_second": 8.276,
"eval_steps_per_second": 0.263,
"step": 226
},
{
"epoch": 2.04,
"learning_rate": 1.7868852459016393e-05,
"loss": 0.0398,
"step": 230
},
{
"epoch": 2.12,
"learning_rate": 1.6229508196721314e-05,
"loss": 0.0419,
"step": 240
},
{
"epoch": 2.21,
"learning_rate": 1.4590163934426229e-05,
"loss": 0.0467,
"step": 250
},
{
"epoch": 2.3,
"learning_rate": 1.2950819672131149e-05,
"loss": 0.0502,
"step": 260
},
{
"epoch": 2.39,
"learning_rate": 1.1311475409836065e-05,
"loss": 0.0304,
"step": 270
},
{
"epoch": 2.48,
"learning_rate": 9.672131147540984e-06,
"loss": 0.0442,
"step": 280
},
{
"epoch": 2.57,
"learning_rate": 8.032786885245902e-06,
"loss": 0.0396,
"step": 290
},
{
"epoch": 2.65,
"learning_rate": 6.393442622950819e-06,
"loss": 0.0429,
"step": 300
},
{
"epoch": 2.74,
"learning_rate": 4.754098360655738e-06,
"loss": 0.0387,
"step": 310
},
{
"epoch": 2.83,
"learning_rate": 3.114754098360656e-06,
"loss": 0.0537,
"step": 320
},
{
"epoch": 2.92,
"learning_rate": 1.4754098360655739e-06,
"loss": 0.0312,
"step": 330
},
{
"epoch": 3.0,
"eval_accuracy": 0.9981308411214953,
"eval_f1": 0.9985111662531018,
"eval_loss": 0.0054078237153589725,
"eval_precision": 0.998015873015873,
"eval_recall": 0.9990069513406157,
"eval_runtime": 193.8262,
"eval_samples_per_second": 8.281,
"eval_steps_per_second": 0.263,
"step": 339
},
{
"epoch": 3.0,
"step": 339,
"total_flos": 3.3948696164738826e+18,
"train_loss": 0.1033598997775784,
"train_runtime": 25716.3258,
"train_samples_per_second": 1.685,
"train_steps_per_second": 0.013
}
],
"max_steps": 339,
"num_train_epochs": 3,
"total_flos": 3.3948696164738826e+18,
"trial_name": null,
"trial_params": null
}