{ "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 }