heroza's picture
End of training
43ef69b verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 23.094688221709006,
"eval_steps": 500,
"global_step": 10000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.23,
"learning_rate": 5.9401e-05,
"loss": 0.3181,
"step": 100
},
{
"epoch": 0.46,
"learning_rate": 5.8802000000000004e-05,
"loss": 0.149,
"step": 200
},
{
"epoch": 0.69,
"learning_rate": 5.8203e-05,
"loss": 0.0917,
"step": 300
},
{
"epoch": 0.92,
"learning_rate": 5.7604e-05,
"loss": 0.0759,
"step": 400
},
{
"epoch": 1.0,
"eval_accuracy_background": 0.9828615415666171,
"eval_accuracy_biofilm": 0.9650712037505778,
"eval_iou_background": 0.980737395917761,
"eval_iou_biofilm": 0.7560937377942886,
"eval_loss": 0.06491132825613022,
"eval_mean_accuracy": 0.9739663726585974,
"eval_mean_iou": 0.8684155668560247,
"eval_overall_accuracy": 0.981822807602889,
"eval_runtime": 59.9845,
"eval_samples_per_second": 10.203,
"eval_steps_per_second": 1.284,
"step": 433
},
{
"epoch": 1.15,
"learning_rate": 5.7005e-05,
"loss": 0.0566,
"step": 500
},
{
"epoch": 1.39,
"learning_rate": 5.6406e-05,
"loss": 0.0504,
"step": 600
},
{
"epoch": 1.62,
"learning_rate": 5.5806999999999996e-05,
"loss": 0.0468,
"step": 700
},
{
"epoch": 1.85,
"learning_rate": 5.5208000000000004e-05,
"loss": 0.0435,
"step": 800
},
{
"epoch": 2.0,
"eval_accuracy_background": 0.9886900685144177,
"eval_accuracy_biofilm": 0.9344582131930536,
"eval_iou_background": 0.9846881804227252,
"eval_iou_biofilm": 0.7903099298609719,
"eval_loss": 0.03914250433444977,
"eval_mean_accuracy": 0.9615741408537356,
"eval_mean_iou": 0.8874990551418486,
"eval_overall_accuracy": 0.9855236041886248,
"eval_runtime": 56.557,
"eval_samples_per_second": 10.821,
"eval_steps_per_second": 1.361,
"step": 866
},
{
"epoch": 2.08,
"learning_rate": 5.4609000000000005e-05,
"loss": 0.0447,
"step": 900
},
{
"epoch": 2.31,
"learning_rate": 5.401e-05,
"loss": 0.0458,
"step": 1000
},
{
"epoch": 2.54,
"learning_rate": 5.3411e-05,
"loss": 0.0348,
"step": 1100
},
{
"epoch": 2.77,
"learning_rate": 5.2812e-05,
"loss": 0.0407,
"step": 1200
},
{
"epoch": 3.0,
"eval_accuracy_background": 0.9887603228091695,
"eval_accuracy_biofilm": 0.9503892545066361,
"eval_iou_background": 0.9857279627494347,
"eval_iou_biofilm": 0.8045544051549277,
"eval_loss": 0.035304486751556396,
"eval_mean_accuracy": 0.9695747886579028,
"eval_mean_iou": 0.8951411839521812,
"eval_overall_accuracy": 0.9865199307332844,
"eval_runtime": 54.7086,
"eval_samples_per_second": 11.187,
"eval_steps_per_second": 1.407,
"step": 1299
},
{
"epoch": 3.0,
"learning_rate": 5.2213e-05,
"loss": 0.0336,
"step": 1300
},
{
"epoch": 3.23,
"learning_rate": 5.1614000000000004e-05,
"loss": 0.0313,
"step": 1400
},
{
"epoch": 3.46,
"learning_rate": 5.1015e-05,
"loss": 0.0338,
"step": 1500
},
{
"epoch": 3.7,
"learning_rate": 5.0416e-05,
"loss": 0.0375,
"step": 1600
},
{
"epoch": 3.93,
"learning_rate": 4.9817e-05,
"loss": 0.0372,
"step": 1700
},
{
"epoch": 4.0,
"eval_accuracy_background": 0.9832945045568867,
"eval_accuracy_biofilm": 0.9788046133867453,
"eval_iou_background": 0.9820038724017046,
"eval_iou_biofilm": 0.7710713706092136,
"eval_loss": 0.048879224807024,
"eval_mean_accuracy": 0.981049558971816,
"eval_mean_iou": 0.8765376215054591,
"eval_overall_accuracy": 0.9830323508811052,
"eval_runtime": 53.641,
"eval_samples_per_second": 11.409,
"eval_steps_per_second": 1.435,
"step": 1732
},
{
"epoch": 4.16,
"learning_rate": 4.9218e-05,
"loss": 0.0362,
"step": 1800
},
{
"epoch": 4.39,
"learning_rate": 4.8619e-05,
"loss": 0.0329,
"step": 1900
},
{
"epoch": 4.62,
"learning_rate": 4.8020000000000004e-05,
"loss": 0.0321,
"step": 2000
},
{
"epoch": 4.85,
"learning_rate": 4.7421000000000006e-05,
"loss": 0.0378,
"step": 2100
},
{
"epoch": 5.0,
"eval_accuracy_background": 0.9919392085185956,
"eval_accuracy_biofilm": 0.9229355314419034,
"eval_iou_background": 0.9872216553975541,
"eval_iou_biofilm": 0.8167600506475522,
"eval_loss": 0.03106585703790188,
"eval_mean_accuracy": 0.9574373699802495,
"eval_mean_iou": 0.9019908530225531,
"eval_overall_accuracy": 0.9879102539398317,
"eval_runtime": 53.6924,
"eval_samples_per_second": 11.398,
"eval_steps_per_second": 1.434,
"step": 2165
},
{
"epoch": 5.08,
"learning_rate": 4.6822e-05,
"loss": 0.0306,
"step": 2200
},
{
"epoch": 5.31,
"learning_rate": 4.6223e-05,
"loss": 0.028,
"step": 2300
},
{
"epoch": 5.54,
"learning_rate": 4.5624e-05,
"loss": 0.0315,
"step": 2400
},
{
"epoch": 5.77,
"learning_rate": 4.5025000000000003e-05,
"loss": 0.0325,
"step": 2500
},
{
"epoch": 6.0,
"eval_accuracy_background": 0.9823462728892177,
"eval_accuracy_biofilm": 0.9665873924240089,
"eval_iou_background": 0.9803152068358649,
"eval_iou_biofilm": 0.7523833558585936,
"eval_loss": 0.051038164645433426,
"eval_mean_accuracy": 0.9744668326566133,
"eval_mean_iou": 0.8663492813472293,
"eval_overall_accuracy": 0.9814261507117296,
"eval_runtime": 53.252,
"eval_samples_per_second": 11.493,
"eval_steps_per_second": 1.446,
"step": 2598
},
{
"epoch": 6.0,
"learning_rate": 4.4426000000000005e-05,
"loss": 0.0389,
"step": 2600
},
{
"epoch": 6.24,
"learning_rate": 4.3827e-05,
"loss": 0.0322,
"step": 2700
},
{
"epoch": 6.47,
"learning_rate": 4.3228e-05,
"loss": 0.0268,
"step": 2800
},
{
"epoch": 6.7,
"learning_rate": 4.2629e-05,
"loss": 0.028,
"step": 2900
},
{
"epoch": 6.93,
"learning_rate": 4.203e-05,
"loss": 0.0306,
"step": 3000
},
{
"epoch": 7.0,
"eval_accuracy_background": 0.9862434447863753,
"eval_accuracy_biofilm": 0.9657386701296414,
"eval_iou_background": 0.9841526345692085,
"eval_iou_biofilm": 0.7903897627132261,
"eval_loss": 0.04278487339615822,
"eval_mean_accuracy": 0.9759910574580084,
"eval_mean_iou": 0.8872711986412174,
"eval_overall_accuracy": 0.9850462215478519,
"eval_runtime": 53.1423,
"eval_samples_per_second": 11.516,
"eval_steps_per_second": 1.449,
"step": 3031
},
{
"epoch": 7.16,
"learning_rate": 4.1431e-05,
"loss": 0.0292,
"step": 3100
},
{
"epoch": 7.39,
"learning_rate": 4.0832e-05,
"loss": 0.0308,
"step": 3200
},
{
"epoch": 7.62,
"learning_rate": 4.0233e-05,
"loss": 0.0268,
"step": 3300
},
{
"epoch": 7.85,
"learning_rate": 3.9634e-05,
"loss": 0.0318,
"step": 3400
},
{
"epoch": 8.0,
"eval_accuracy_background": 0.9859234209489122,
"eval_accuracy_biofilm": 0.9617940703893646,
"eval_iou_background": 0.983593215391998,
"eval_iou_biofilm": 0.7838504582636022,
"eval_loss": 0.03987625986337662,
"eval_mean_accuracy": 0.9738587456691383,
"eval_mean_iou": 0.8837218368278001,
"eval_overall_accuracy": 0.9845145676596673,
"eval_runtime": 52.5703,
"eval_samples_per_second": 11.642,
"eval_steps_per_second": 1.465,
"step": 3464
},
{
"epoch": 8.08,
"learning_rate": 3.9035e-05,
"loss": 0.0269,
"step": 3500
},
{
"epoch": 8.31,
"learning_rate": 3.8436e-05,
"loss": 0.0321,
"step": 3600
},
{
"epoch": 8.55,
"learning_rate": 3.7837000000000004e-05,
"loss": 0.0305,
"step": 3700
},
{
"epoch": 8.78,
"learning_rate": 3.7238000000000005e-05,
"loss": 0.0302,
"step": 3800
},
{
"epoch": 9.0,
"eval_accuracy_background": 0.9859386046114496,
"eval_accuracy_biofilm": 0.9519521052979112,
"eval_iou_background": 0.9830098689255221,
"eval_iou_biofilm": 0.7759842341679627,
"eval_loss": 0.043639492243528366,
"eval_mean_accuracy": 0.9689453549546804,
"eval_mean_iou": 0.8794970515467424,
"eval_overall_accuracy": 0.9839542167088564,
"eval_runtime": 53.0115,
"eval_samples_per_second": 11.545,
"eval_steps_per_second": 1.453,
"step": 3897
},
{
"epoch": 9.01,
"learning_rate": 3.6639e-05,
"loss": 0.0243,
"step": 3900
},
{
"epoch": 9.24,
"learning_rate": 3.604e-05,
"loss": 0.029,
"step": 4000
},
{
"epoch": 9.47,
"learning_rate": 3.544100000000001e-05,
"loss": 0.0279,
"step": 4100
},
{
"epoch": 9.7,
"learning_rate": 3.4842e-05,
"loss": 0.0312,
"step": 4200
},
{
"epoch": 9.93,
"learning_rate": 3.4243000000000004e-05,
"loss": 0.0236,
"step": 4300
},
{
"epoch": 10.0,
"eval_accuracy_background": 0.9866936591469345,
"eval_accuracy_biofilm": 0.9559968525080889,
"eval_iou_background": 0.9840087435512237,
"eval_iou_biofilm": 0.7870938963524574,
"eval_loss": 0.03914466127753258,
"eval_mean_accuracy": 0.9713452558275117,
"eval_mean_iou": 0.8855513199518406,
"eval_overall_accuracy": 0.9849013482890138,
"eval_runtime": 52.0282,
"eval_samples_per_second": 11.763,
"eval_steps_per_second": 1.48,
"step": 4330
},
{
"epoch": 10.16,
"learning_rate": 3.3644000000000005e-05,
"loss": 0.0246,
"step": 4400
},
{
"epoch": 10.39,
"learning_rate": 3.3045000000000006e-05,
"loss": 0.0235,
"step": 4500
},
{
"epoch": 10.62,
"learning_rate": 3.2446e-05,
"loss": 0.0268,
"step": 4600
},
{
"epoch": 10.85,
"learning_rate": 3.1847e-05,
"loss": 0.0247,
"step": 4700
},
{
"epoch": 11.0,
"eval_accuracy_background": 0.9834379952859539,
"eval_accuracy_biofilm": 0.9627888319063236,
"eval_iou_background": 0.9811740428860415,
"eval_iou_biofilm": 0.7598397926761448,
"eval_loss": 0.0451015830039978,
"eval_mean_accuracy": 0.9731134135961388,
"eval_mean_iou": 0.8705069177810931,
"eval_overall_accuracy": 0.9822323415458852,
"eval_runtime": 52.2459,
"eval_samples_per_second": 11.714,
"eval_steps_per_second": 1.474,
"step": 4763
},
{
"epoch": 11.09,
"learning_rate": 3.1248e-05,
"loss": 0.023,
"step": 4800
},
{
"epoch": 11.32,
"learning_rate": 3.0649000000000004e-05,
"loss": 0.0242,
"step": 4900
},
{
"epoch": 11.55,
"learning_rate": 3.0050000000000002e-05,
"loss": 0.0263,
"step": 5000
},
{
"epoch": 11.78,
"learning_rate": 2.9451e-05,
"loss": 0.0213,
"step": 5100
},
{
"epoch": 12.0,
"eval_accuracy_background": 0.9823563316391414,
"eval_accuracy_biofilm": 0.9646785376268351,
"eval_iou_background": 0.9802094628384811,
"eval_iou_biofilm": 0.7509923460988596,
"eval_loss": 0.04865105450153351,
"eval_mean_accuracy": 0.9735174346329882,
"eval_mean_iou": 0.8656009044686703,
"eval_overall_accuracy": 0.9813241688306411,
"eval_runtime": 52.4181,
"eval_samples_per_second": 11.675,
"eval_steps_per_second": 1.469,
"step": 5196
},
{
"epoch": 12.01,
"learning_rate": 2.8851999999999998e-05,
"loss": 0.0309,
"step": 5200
},
{
"epoch": 12.24,
"learning_rate": 2.8253e-05,
"loss": 0.0252,
"step": 5300
},
{
"epoch": 12.47,
"learning_rate": 2.7653999999999996e-05,
"loss": 0.0234,
"step": 5400
},
{
"epoch": 12.7,
"learning_rate": 2.7054999999999998e-05,
"loss": 0.0235,
"step": 5500
},
{
"epoch": 12.93,
"learning_rate": 2.6455999999999995e-05,
"loss": 0.0256,
"step": 5600
},
{
"epoch": 13.0,
"eval_accuracy_background": 0.9863652279961024,
"eval_accuracy_biofilm": 0.9473298043272511,
"eval_iou_background": 0.9831542706255657,
"eval_iou_biofilm": 0.7765716428707975,
"eval_loss": 0.0444067120552063,
"eval_mean_accuracy": 0.9668475161616767,
"eval_mean_iou": 0.8798629567481816,
"eval_overall_accuracy": 0.9840860458483189,
"eval_runtime": 52.5359,
"eval_samples_per_second": 11.649,
"eval_steps_per_second": 1.466,
"step": 5629
},
{
"epoch": 13.16,
"learning_rate": 2.5857e-05,
"loss": 0.0228,
"step": 5700
},
{
"epoch": 13.39,
"learning_rate": 2.5258e-05,
"loss": 0.0255,
"step": 5800
},
{
"epoch": 13.63,
"learning_rate": 2.4659e-05,
"loss": 0.0212,
"step": 5900
},
{
"epoch": 13.86,
"learning_rate": 2.406e-05,
"loss": 0.0218,
"step": 6000
},
{
"epoch": 14.0,
"eval_accuracy_background": 0.9821700673384857,
"eval_accuracy_biofilm": 0.9724741927673717,
"eval_iou_background": 0.9804965351674374,
"eval_iou_biofilm": 0.7552949378009881,
"eval_loss": 0.049166884273290634,
"eval_mean_accuracy": 0.9773221300529287,
"eval_mean_iou": 0.8678957364842128,
"eval_overall_accuracy": 0.9816039491270605,
"eval_runtime": 52.7983,
"eval_samples_per_second": 11.591,
"eval_steps_per_second": 1.458,
"step": 6062
},
{
"epoch": 14.09,
"learning_rate": 2.3460999999999998e-05,
"loss": 0.0297,
"step": 6100
},
{
"epoch": 14.32,
"learning_rate": 2.2862e-05,
"loss": 0.0286,
"step": 6200
},
{
"epoch": 14.55,
"learning_rate": 2.2263e-05,
"loss": 0.0219,
"step": 6300
},
{
"epoch": 14.78,
"learning_rate": 2.1663999999999998e-05,
"loss": 0.0216,
"step": 6400
},
{
"epoch": 15.0,
"eval_accuracy_background": 0.983388424892843,
"eval_accuracy_biofilm": 0.9662580062949838,
"eval_iou_background": 0.981335202197703,
"eval_iou_biofilm": 0.7620968774425835,
"eval_loss": 0.05022722855210304,
"eval_mean_accuracy": 0.9748232155939134,
"eval_mean_iou": 0.8717160398201432,
"eval_overall_accuracy": 0.9823882219852604,
"eval_runtime": 52.3676,
"eval_samples_per_second": 11.687,
"eval_steps_per_second": 1.47,
"step": 6495
},
{
"epoch": 15.01,
"learning_rate": 2.1065e-05,
"loss": 0.022,
"step": 6500
},
{
"epoch": 15.24,
"learning_rate": 2.0465999999999997e-05,
"loss": 0.0211,
"step": 6600
},
{
"epoch": 15.47,
"learning_rate": 1.9866999999999998e-05,
"loss": 0.0236,
"step": 6700
},
{
"epoch": 15.7,
"learning_rate": 1.9267999999999996e-05,
"loss": 0.0225,
"step": 6800
},
{
"epoch": 15.94,
"learning_rate": 1.8669e-05,
"loss": 0.0206,
"step": 6900
},
{
"epoch": 16.0,
"eval_accuracy_background": 0.9811236366128779,
"eval_accuracy_biofilm": 0.9721258776660137,
"eval_iou_background": 0.9794307694409038,
"eval_iou_biofilm": 0.7452563861896737,
"eval_loss": 0.056503016501665115,
"eval_mean_accuracy": 0.9766247571394457,
"eval_mean_iou": 0.8623435778152888,
"eval_overall_accuracy": 0.9805982796515773,
"eval_runtime": 53.0323,
"eval_samples_per_second": 11.54,
"eval_steps_per_second": 1.452,
"step": 6928
},
{
"epoch": 16.17,
"learning_rate": 1.807e-05,
"loss": 0.022,
"step": 7000
},
{
"epoch": 16.4,
"learning_rate": 1.7471e-05,
"loss": 0.0208,
"step": 7100
},
{
"epoch": 16.63,
"learning_rate": 1.6872e-05,
"loss": 0.0234,
"step": 7200
},
{
"epoch": 16.86,
"learning_rate": 1.6272999999999998e-05,
"loss": 0.0223,
"step": 7300
},
{
"epoch": 17.0,
"eval_accuracy_background": 0.9826374580040367,
"eval_accuracy_biofilm": 0.963460810424141,
"eval_iou_background": 0.9804161065014417,
"eval_iou_biofilm": 0.7527009753594974,
"eval_loss": 0.0508936271071434,
"eval_mean_accuracy": 0.9730491342140888,
"eval_mean_iou": 0.8665585409304696,
"eval_overall_accuracy": 0.981517780831087,
"eval_runtime": 53.2214,
"eval_samples_per_second": 11.499,
"eval_steps_per_second": 1.447,
"step": 7361
},
{
"epoch": 17.09,
"learning_rate": 1.5674e-05,
"loss": 0.0247,
"step": 7400
},
{
"epoch": 17.32,
"learning_rate": 1.5075000000000002e-05,
"loss": 0.022,
"step": 7500
},
{
"epoch": 17.55,
"learning_rate": 1.4476e-05,
"loss": 0.0265,
"step": 7600
},
{
"epoch": 17.78,
"learning_rate": 1.3877e-05,
"loss": 0.0226,
"step": 7700
},
{
"epoch": 18.0,
"eval_accuracy_background": 0.9841817422579221,
"eval_accuracy_biofilm": 0.9742679330002422,
"eval_iou_background": 0.9826138886496493,
"eval_iou_biofilm": 0.7762471663828775,
"eval_loss": 0.04640285670757294,
"eval_mean_accuracy": 0.9792248376290822,
"eval_mean_iou": 0.8794305275162634,
"eval_overall_accuracy": 0.9836028993774575,
"eval_runtime": 53.7901,
"eval_samples_per_second": 11.378,
"eval_steps_per_second": 1.431,
"step": 7794
},
{
"epoch": 18.01,
"learning_rate": 1.3277999999999999e-05,
"loss": 0.0189,
"step": 7800
},
{
"epoch": 18.24,
"learning_rate": 1.2678999999999998e-05,
"loss": 0.0196,
"step": 7900
},
{
"epoch": 18.48,
"learning_rate": 1.2079999999999998e-05,
"loss": 0.0239,
"step": 8000
},
{
"epoch": 18.71,
"learning_rate": 1.1480999999999997e-05,
"loss": 0.0224,
"step": 8100
},
{
"epoch": 18.94,
"learning_rate": 1.0882000000000004e-05,
"loss": 0.0243,
"step": 8200
},
{
"epoch": 19.0,
"eval_accuracy_background": 0.9806471221015685,
"eval_accuracy_biofilm": 0.9842518653841921,
"eval_iou_background": 0.9796904437901697,
"eval_iou_biofilm": 0.7501332139181935,
"eval_loss": 0.05462173372507095,
"eval_mean_accuracy": 0.9824494937428803,
"eval_mean_iou": 0.8649118288541816,
"eval_overall_accuracy": 0.9808575941744937,
"eval_runtime": 53.6405,
"eval_samples_per_second": 11.409,
"eval_steps_per_second": 1.435,
"step": 8227
},
{
"epoch": 19.17,
"learning_rate": 1.0283000000000003e-05,
"loss": 0.0201,
"step": 8300
},
{
"epoch": 19.4,
"learning_rate": 9.684000000000002e-06,
"loss": 0.0203,
"step": 8400
},
{
"epoch": 19.63,
"learning_rate": 9.085000000000002e-06,
"loss": 0.017,
"step": 8500
},
{
"epoch": 19.86,
"learning_rate": 8.486000000000001e-06,
"loss": 0.02,
"step": 8600
},
{
"epoch": 20.0,
"eval_accuracy_background": 0.9816362984772036,
"eval_accuracy_biofilm": 0.9715923447714216,
"eval_iou_background": 0.9799101839896647,
"eval_iou_biofilm": 0.7495984750713218,
"eval_loss": 0.05672093480825424,
"eval_mean_accuracy": 0.9766143216243126,
"eval_mean_iou": 0.8647543295304932,
"eval_overall_accuracy": 0.98104985678328,
"eval_runtime": 52.5058,
"eval_samples_per_second": 11.656,
"eval_steps_per_second": 1.467,
"step": 8660
},
{
"epoch": 20.09,
"learning_rate": 7.887000000000001e-06,
"loss": 0.0213,
"step": 8700
},
{
"epoch": 20.32,
"learning_rate": 7.2879999999999995e-06,
"loss": 0.0218,
"step": 8800
},
{
"epoch": 20.55,
"learning_rate": 6.688999999999999e-06,
"loss": 0.0233,
"step": 8900
},
{
"epoch": 20.79,
"learning_rate": 6.0899999999999984e-06,
"loss": 0.0196,
"step": 9000
},
{
"epoch": 21.0,
"eval_accuracy_background": 0.9813258625255226,
"eval_accuracy_biofilm": 0.9755059098012457,
"eval_iou_background": 0.9798376554115104,
"eval_iou_biofilm": 0.7497220481513676,
"eval_loss": 0.05593876540660858,
"eval_mean_accuracy": 0.9784158861633842,
"eval_mean_iou": 0.8647798517814389,
"eval_overall_accuracy": 0.9809860498348075,
"eval_runtime": 53.7718,
"eval_samples_per_second": 11.381,
"eval_steps_per_second": 1.432,
"step": 9093
},
{
"epoch": 21.02,
"learning_rate": 5.490999999999998e-06,
"loss": 0.0201,
"step": 9100
},
{
"epoch": 21.25,
"learning_rate": 4.891999999999997e-06,
"loss": 0.0167,
"step": 9200
},
{
"epoch": 21.48,
"learning_rate": 4.292999999999997e-06,
"loss": 0.0199,
"step": 9300
},
{
"epoch": 21.71,
"learning_rate": 3.694000000000003e-06,
"loss": 0.0225,
"step": 9400
},
{
"epoch": 21.94,
"learning_rate": 3.0950000000000026e-06,
"loss": 0.0206,
"step": 9500
},
{
"epoch": 22.0,
"eval_accuracy_background": 0.9815057872205616,
"eval_accuracy_biofilm": 0.9742263332819756,
"eval_iou_background": 0.9799396728065195,
"eval_iou_biofilm": 0.7504120847099512,
"eval_loss": 0.05521002784371376,
"eval_mean_accuracy": 0.9778660602512685,
"eval_mean_iou": 0.8651758787582353,
"eval_overall_accuracy": 0.9810807578522835,
"eval_runtime": 44.0076,
"eval_samples_per_second": 13.907,
"eval_steps_per_second": 1.75,
"step": 9526
},
{
"epoch": 22.17,
"learning_rate": 2.496000000000002e-06,
"loss": 0.0191,
"step": 9600
},
{
"epoch": 22.4,
"learning_rate": 1.8970000000000013e-06,
"loss": 0.0162,
"step": 9700
},
{
"epoch": 22.63,
"learning_rate": 1.298000000000001e-06,
"loss": 0.0213,
"step": 9800
},
{
"epoch": 22.86,
"learning_rate": 6.990000000000005e-07,
"loss": 0.0189,
"step": 9900
},
{
"epoch": 23.0,
"eval_accuracy_background": 0.9815976262005296,
"eval_accuracy_biofilm": 0.9753409416063213,
"eval_iou_background": 0.9800989963754594,
"eval_iou_biofilm": 0.7521286753495444,
"eval_loss": 0.05435158312320709,
"eval_mean_accuracy": 0.9784692839034255,
"eval_mean_iou": 0.8661138358625019,
"eval_overall_accuracy": 0.9812323138126737,
"eval_runtime": 44.1254,
"eval_samples_per_second": 13.87,
"eval_steps_per_second": 1.745,
"step": 9959
},
{
"epoch": 23.09,
"learning_rate": 1e-07,
"loss": 0.0208,
"step": 10000
},
{
"epoch": 23.09,
"eval_accuracy_background": 0.9820080627772998,
"eval_accuracy_biofilm": 0.973973103250941,
"eval_iou_background": 0.9804257765494778,
"eval_iou_biofilm": 0.7549272284294235,
"eval_loss": 0.05298486724495888,
"eval_mean_accuracy": 0.9779905830141205,
"eval_mean_iou": 0.8676765024894506,
"eval_overall_accuracy": 0.9815389213004198,
"eval_runtime": 44.7948,
"eval_samples_per_second": 13.662,
"eval_steps_per_second": 1.719,
"step": 10000
},
{
"epoch": 23.09,
"step": 10000,
"total_flos": 1.4014301557688893e+18,
"train_loss": 0.03242814379930496,
"train_runtime": 14425.5946,
"train_samples_per_second": 5.546,
"train_steps_per_second": 0.693
}
],
"logging_steps": 100,
"max_steps": 10000,
"num_input_tokens_seen": 0,
"num_train_epochs": 24,
"save_steps": 500,
"total_flos": 1.4014301557688893e+18,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}