|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 35.714285714285715, |
|
"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.36, |
|
"learning_rate": 5.9401e-05, |
|
"loss": 0.3125, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 5.8802000000000004e-05, |
|
"loss": 0.1622, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.942847215481601, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.942847215481601, |
|
"eval_loss": 0.11576633900403976, |
|
"eval_mean_accuracy": 0.942847215481601, |
|
"eval_mean_iou": 0.4714236077408005, |
|
"eval_overall_accuracy": 0.942847215481601, |
|
"eval_runtime": 526.3881, |
|
"eval_samples_per_second": 1.066, |
|
"eval_steps_per_second": 0.135, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 1.07, |
|
"learning_rate": 5.8203e-05, |
|
"loss": 0.1116, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 1.43, |
|
"learning_rate": 5.7604e-05, |
|
"loss": 0.0793, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 1.79, |
|
"learning_rate": 5.7005e-05, |
|
"loss": 0.0742, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9090055653431761, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9090055653431761, |
|
"eval_loss": 0.06430274248123169, |
|
"eval_mean_accuracy": 0.9090055653431761, |
|
"eval_mean_iou": 0.45450278267158806, |
|
"eval_overall_accuracy": 0.9090055653431761, |
|
"eval_runtime": 543.6769, |
|
"eval_samples_per_second": 1.032, |
|
"eval_steps_per_second": 0.131, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"learning_rate": 5.6406e-05, |
|
"loss": 0.0636, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 2.5, |
|
"learning_rate": 5.5806999999999996e-05, |
|
"loss": 0.0573, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 2.86, |
|
"learning_rate": 5.5208000000000004e-05, |
|
"loss": 0.0549, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9594193798323095, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9594193798323095, |
|
"eval_loss": 0.05815902352333069, |
|
"eval_mean_accuracy": 0.9594193798323095, |
|
"eval_mean_iou": 0.47970968991615476, |
|
"eval_overall_accuracy": 0.9594193798323095, |
|
"eval_runtime": 547.5124, |
|
"eval_samples_per_second": 1.025, |
|
"eval_steps_per_second": 0.13, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 3.21, |
|
"learning_rate": 5.4609000000000005e-05, |
|
"loss": 0.0476, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 3.57, |
|
"learning_rate": 5.401e-05, |
|
"loss": 0.0534, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 3.93, |
|
"learning_rate": 5.3411e-05, |
|
"loss": 0.0459, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.947494221199843, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.947494221199843, |
|
"eval_loss": 0.050838589668273926, |
|
"eval_mean_accuracy": 0.947494221199843, |
|
"eval_mean_iou": 0.4737471105999215, |
|
"eval_overall_accuracy": 0.947494221199843, |
|
"eval_runtime": 550.0782, |
|
"eval_samples_per_second": 1.02, |
|
"eval_steps_per_second": 0.129, |
|
"step": 1120 |
|
}, |
|
{ |
|
"epoch": 4.29, |
|
"learning_rate": 5.2812e-05, |
|
"loss": 0.0505, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 4.64, |
|
"learning_rate": 5.2213e-05, |
|
"loss": 0.0408, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"learning_rate": 5.1614000000000004e-05, |
|
"loss": 0.0506, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9410741911182998, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9410741911182998, |
|
"eval_loss": 0.04050213098526001, |
|
"eval_mean_accuracy": 0.9410741911182998, |
|
"eval_mean_iou": 0.4705370955591499, |
|
"eval_overall_accuracy": 0.9410741911182998, |
|
"eval_runtime": 552.6887, |
|
"eval_samples_per_second": 1.015, |
|
"eval_steps_per_second": 0.128, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 5.36, |
|
"learning_rate": 5.1015e-05, |
|
"loss": 0.0424, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 5.71, |
|
"learning_rate": 5.0416e-05, |
|
"loss": 0.0411, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 6.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9729236047343641, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9729236047343641, |
|
"eval_loss": 0.04759969189763069, |
|
"eval_mean_accuracy": 0.9729236047343641, |
|
"eval_mean_iou": 0.48646180236718206, |
|
"eval_overall_accuracy": 0.9729236047343641, |
|
"eval_runtime": 550.2065, |
|
"eval_samples_per_second": 1.02, |
|
"eval_steps_per_second": 0.129, |
|
"step": 1680 |
|
}, |
|
{ |
|
"epoch": 6.07, |
|
"learning_rate": 4.9817e-05, |
|
"loss": 0.0447, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 6.43, |
|
"learning_rate": 4.9218e-05, |
|
"loss": 0.0373, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 6.79, |
|
"learning_rate": 4.8619e-05, |
|
"loss": 0.0456, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 7.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9508669273962801, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9508669273962801, |
|
"eval_loss": 0.047635335475206375, |
|
"eval_mean_accuracy": 0.9508669273962801, |
|
"eval_mean_iou": 0.47543346369814005, |
|
"eval_overall_accuracy": 0.9508669273962801, |
|
"eval_runtime": 546.2221, |
|
"eval_samples_per_second": 1.027, |
|
"eval_steps_per_second": 0.13, |
|
"step": 1960 |
|
}, |
|
{ |
|
"epoch": 7.14, |
|
"learning_rate": 4.8020000000000004e-05, |
|
"loss": 0.043, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 7.5, |
|
"learning_rate": 4.7421000000000006e-05, |
|
"loss": 0.0368, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 7.86, |
|
"learning_rate": 4.6822e-05, |
|
"loss": 0.0381, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9584158313784339, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9584158313784339, |
|
"eval_loss": 0.05542508885264397, |
|
"eval_mean_accuracy": 0.9584158313784339, |
|
"eval_mean_iou": 0.47920791568921695, |
|
"eval_overall_accuracy": 0.9584158313784339, |
|
"eval_runtime": 550.321, |
|
"eval_samples_per_second": 1.019, |
|
"eval_steps_per_second": 0.129, |
|
"step": 2240 |
|
}, |
|
{ |
|
"epoch": 8.21, |
|
"learning_rate": 4.6223e-05, |
|
"loss": 0.043, |
|
"step": 2300 |
|
}, |
|
{ |
|
"epoch": 8.57, |
|
"learning_rate": 4.5624e-05, |
|
"loss": 0.0343, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 8.93, |
|
"learning_rate": 4.5025000000000003e-05, |
|
"loss": 0.0348, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 9.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9778518410986721, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9778518410986721, |
|
"eval_loss": 0.055933739989995956, |
|
"eval_mean_accuracy": 0.9778518410986721, |
|
"eval_mean_iou": 0.48892592054933603, |
|
"eval_overall_accuracy": 0.9778518410986721, |
|
"eval_runtime": 549.6085, |
|
"eval_samples_per_second": 1.021, |
|
"eval_steps_per_second": 0.129, |
|
"step": 2520 |
|
}, |
|
{ |
|
"epoch": 9.29, |
|
"learning_rate": 4.4426000000000005e-05, |
|
"loss": 0.0386, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 9.64, |
|
"learning_rate": 4.3827e-05, |
|
"loss": 0.0357, |
|
"step": 2700 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"learning_rate": 4.3228e-05, |
|
"loss": 0.0388, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9514310207538597, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9514310207538597, |
|
"eval_loss": 0.05132460966706276, |
|
"eval_mean_accuracy": 0.9514310207538597, |
|
"eval_mean_iou": 0.47571551037692983, |
|
"eval_overall_accuracy": 0.9514310207538597, |
|
"eval_runtime": 564.8695, |
|
"eval_samples_per_second": 0.993, |
|
"eval_steps_per_second": 0.126, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 10.36, |
|
"learning_rate": 4.2629e-05, |
|
"loss": 0.0343, |
|
"step": 2900 |
|
}, |
|
{ |
|
"epoch": 10.71, |
|
"learning_rate": 4.203e-05, |
|
"loss": 0.0385, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 11.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9766834540345947, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9766834540345947, |
|
"eval_loss": 0.06599694490432739, |
|
"eval_mean_accuracy": 0.9766834540345947, |
|
"eval_mean_iou": 0.48834172701729733, |
|
"eval_overall_accuracy": 0.9766834540345947, |
|
"eval_runtime": 550.984, |
|
"eval_samples_per_second": 1.018, |
|
"eval_steps_per_second": 0.129, |
|
"step": 3080 |
|
}, |
|
{ |
|
"epoch": 11.07, |
|
"learning_rate": 4.1431e-05, |
|
"loss": 0.0369, |
|
"step": 3100 |
|
}, |
|
{ |
|
"epoch": 11.43, |
|
"learning_rate": 4.0832e-05, |
|
"loss": 0.0373, |
|
"step": 3200 |
|
}, |
|
{ |
|
"epoch": 11.79, |
|
"learning_rate": 4.0233e-05, |
|
"loss": 0.0309, |
|
"step": 3300 |
|
}, |
|
{ |
|
"epoch": 12.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9615736203248841, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9615736203248841, |
|
"eval_loss": 0.05894804745912552, |
|
"eval_mean_accuracy": 0.9615736203248841, |
|
"eval_mean_iou": 0.48078681016244207, |
|
"eval_overall_accuracy": 0.9615736203248841, |
|
"eval_runtime": 552.1563, |
|
"eval_samples_per_second": 1.016, |
|
"eval_steps_per_second": 0.129, |
|
"step": 3360 |
|
}, |
|
{ |
|
"epoch": 12.14, |
|
"learning_rate": 3.9634e-05, |
|
"loss": 0.0339, |
|
"step": 3400 |
|
}, |
|
{ |
|
"epoch": 12.5, |
|
"learning_rate": 3.9035e-05, |
|
"loss": 0.0386, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 12.86, |
|
"learning_rate": 3.8436e-05, |
|
"loss": 0.0322, |
|
"step": 3600 |
|
}, |
|
{ |
|
"epoch": 13.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9591686430406582, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9591686430406582, |
|
"eval_loss": 0.05389216169714928, |
|
"eval_mean_accuracy": 0.9591686430406582, |
|
"eval_mean_iou": 0.4795843215203291, |
|
"eval_overall_accuracy": 0.9591686430406582, |
|
"eval_runtime": 553.0371, |
|
"eval_samples_per_second": 1.014, |
|
"eval_steps_per_second": 0.128, |
|
"step": 3640 |
|
}, |
|
{ |
|
"epoch": 13.21, |
|
"learning_rate": 3.7837000000000004e-05, |
|
"loss": 0.0346, |
|
"step": 3700 |
|
}, |
|
{ |
|
"epoch": 13.57, |
|
"learning_rate": 3.7238000000000005e-05, |
|
"loss": 0.0328, |
|
"step": 3800 |
|
}, |
|
{ |
|
"epoch": 13.93, |
|
"learning_rate": 3.6639e-05, |
|
"loss": 0.0361, |
|
"step": 3900 |
|
}, |
|
{ |
|
"epoch": 14.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.962455880546553, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.962455880546553, |
|
"eval_loss": 0.06209651753306389, |
|
"eval_mean_accuracy": 0.962455880546553, |
|
"eval_mean_iou": 0.4812279402732765, |
|
"eval_overall_accuracy": 0.962455880546553, |
|
"eval_runtime": 550.3023, |
|
"eval_samples_per_second": 1.019, |
|
"eval_steps_per_second": 0.129, |
|
"step": 3920 |
|
}, |
|
{ |
|
"epoch": 14.29, |
|
"learning_rate": 3.604e-05, |
|
"loss": 0.0335, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 14.64, |
|
"learning_rate": 3.544100000000001e-05, |
|
"loss": 0.033, |
|
"step": 4100 |
|
}, |
|
{ |
|
"epoch": 15.0, |
|
"learning_rate": 3.4842e-05, |
|
"loss": 0.0277, |
|
"step": 4200 |
|
}, |
|
{ |
|
"epoch": 15.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9672051909989013, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9672051909989013, |
|
"eval_loss": 0.05757427215576172, |
|
"eval_mean_accuracy": 0.9672051909989013, |
|
"eval_mean_iou": 0.48360259549945067, |
|
"eval_overall_accuracy": 0.9672051909989013, |
|
"eval_runtime": 553.625, |
|
"eval_samples_per_second": 1.013, |
|
"eval_steps_per_second": 0.128, |
|
"step": 4200 |
|
}, |
|
{ |
|
"epoch": 15.36, |
|
"learning_rate": 3.4243000000000004e-05, |
|
"loss": 0.0272, |
|
"step": 4300 |
|
}, |
|
{ |
|
"epoch": 15.71, |
|
"learning_rate": 3.3644000000000005e-05, |
|
"loss": 0.0324, |
|
"step": 4400 |
|
}, |
|
{ |
|
"epoch": 16.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9404193033829281, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9404193033829281, |
|
"eval_loss": 0.05032452195882797, |
|
"eval_mean_accuracy": 0.9404193033829281, |
|
"eval_mean_iou": 0.47020965169146406, |
|
"eval_overall_accuracy": 0.9404193033829281, |
|
"eval_runtime": 552.3884, |
|
"eval_samples_per_second": 1.016, |
|
"eval_steps_per_second": 0.129, |
|
"step": 4480 |
|
}, |
|
{ |
|
"epoch": 16.07, |
|
"learning_rate": 3.3045000000000006e-05, |
|
"loss": 0.0329, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 16.43, |
|
"learning_rate": 3.2446e-05, |
|
"loss": 0.0306, |
|
"step": 4600 |
|
}, |
|
{ |
|
"epoch": 16.79, |
|
"learning_rate": 3.1847e-05, |
|
"loss": 0.0355, |
|
"step": 4700 |
|
}, |
|
{ |
|
"epoch": 17.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9601482259019094, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9601482259019094, |
|
"eval_loss": 0.058313556015491486, |
|
"eval_mean_accuracy": 0.9601482259019094, |
|
"eval_mean_iou": 0.4800741129509547, |
|
"eval_overall_accuracy": 0.9601482259019094, |
|
"eval_runtime": 556.5894, |
|
"eval_samples_per_second": 1.008, |
|
"eval_steps_per_second": 0.128, |
|
"step": 4760 |
|
}, |
|
{ |
|
"epoch": 17.14, |
|
"learning_rate": 3.1248e-05, |
|
"loss": 0.0272, |
|
"step": 4800 |
|
}, |
|
{ |
|
"epoch": 17.5, |
|
"learning_rate": 3.0649000000000004e-05, |
|
"loss": 0.0284, |
|
"step": 4900 |
|
}, |
|
{ |
|
"epoch": 17.86, |
|
"learning_rate": 3.0050000000000002e-05, |
|
"loss": 0.032, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 18.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9358022761300121, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9358022761300121, |
|
"eval_loss": 0.05276491492986679, |
|
"eval_mean_accuracy": 0.9358022761300121, |
|
"eval_mean_iou": 0.46790113806500605, |
|
"eval_overall_accuracy": 0.9358022761300121, |
|
"eval_runtime": 555.4257, |
|
"eval_samples_per_second": 1.01, |
|
"eval_steps_per_second": 0.128, |
|
"step": 5040 |
|
}, |
|
{ |
|
"epoch": 18.21, |
|
"learning_rate": 2.9451e-05, |
|
"loss": 0.0299, |
|
"step": 5100 |
|
}, |
|
{ |
|
"epoch": 18.57, |
|
"learning_rate": 2.8851999999999998e-05, |
|
"loss": 0.0312, |
|
"step": 5200 |
|
}, |
|
{ |
|
"epoch": 18.93, |
|
"learning_rate": 2.8253e-05, |
|
"loss": 0.0275, |
|
"step": 5300 |
|
}, |
|
{ |
|
"epoch": 19.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.965586697464028, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.965586697464028, |
|
"eval_loss": 0.06823164969682693, |
|
"eval_mean_accuracy": 0.965586697464028, |
|
"eval_mean_iou": 0.482793348732014, |
|
"eval_overall_accuracy": 0.965586697464028, |
|
"eval_runtime": 549.8075, |
|
"eval_samples_per_second": 1.02, |
|
"eval_steps_per_second": 0.129, |
|
"step": 5320 |
|
}, |
|
{ |
|
"epoch": 19.29, |
|
"learning_rate": 2.7653999999999996e-05, |
|
"loss": 0.0268, |
|
"step": 5400 |
|
}, |
|
{ |
|
"epoch": 19.64, |
|
"learning_rate": 2.7054999999999998e-05, |
|
"loss": 0.0312, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"learning_rate": 2.6455999999999995e-05, |
|
"loss": 0.0329, |
|
"step": 5600 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9591398671498725, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9591398671498725, |
|
"eval_loss": 0.0712042823433876, |
|
"eval_mean_accuracy": 0.9591398671498725, |
|
"eval_mean_iou": 0.47956993357493627, |
|
"eval_overall_accuracy": 0.9591398671498725, |
|
"eval_runtime": 562.937, |
|
"eval_samples_per_second": 0.997, |
|
"eval_steps_per_second": 0.126, |
|
"step": 5600 |
|
}, |
|
{ |
|
"epoch": 20.36, |
|
"learning_rate": 2.5857e-05, |
|
"loss": 0.0248, |
|
"step": 5700 |
|
}, |
|
{ |
|
"epoch": 20.71, |
|
"learning_rate": 2.5258e-05, |
|
"loss": 0.0284, |
|
"step": 5800 |
|
}, |
|
{ |
|
"epoch": 21.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9736688573566209, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9736688573566209, |
|
"eval_loss": 0.07690689712762833, |
|
"eval_mean_accuracy": 0.9736688573566209, |
|
"eval_mean_iou": 0.48683442867831045, |
|
"eval_overall_accuracy": 0.9736688573566209, |
|
"eval_runtime": 556.998, |
|
"eval_samples_per_second": 1.007, |
|
"eval_steps_per_second": 0.127, |
|
"step": 5880 |
|
}, |
|
{ |
|
"epoch": 21.07, |
|
"learning_rate": 2.4659e-05, |
|
"loss": 0.0284, |
|
"step": 5900 |
|
}, |
|
{ |
|
"epoch": 21.43, |
|
"learning_rate": 2.406e-05, |
|
"loss": 0.027, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 21.79, |
|
"learning_rate": 2.3460999999999998e-05, |
|
"loss": 0.028, |
|
"step": 6100 |
|
}, |
|
{ |
|
"epoch": 22.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9651370204841393, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9651370204841393, |
|
"eval_loss": 0.06147192046046257, |
|
"eval_mean_accuracy": 0.9651370204841393, |
|
"eval_mean_iou": 0.48256851024206965, |
|
"eval_overall_accuracy": 0.9651370204841393, |
|
"eval_runtime": 545.2774, |
|
"eval_samples_per_second": 1.029, |
|
"eval_steps_per_second": 0.13, |
|
"step": 6160 |
|
}, |
|
{ |
|
"epoch": 22.14, |
|
"learning_rate": 2.2862e-05, |
|
"loss": 0.0306, |
|
"step": 6200 |
|
}, |
|
{ |
|
"epoch": 22.5, |
|
"learning_rate": 2.2263e-05, |
|
"loss": 0.0296, |
|
"step": 6300 |
|
}, |
|
{ |
|
"epoch": 22.86, |
|
"learning_rate": 2.1663999999999998e-05, |
|
"loss": 0.0275, |
|
"step": 6400 |
|
}, |
|
{ |
|
"epoch": 23.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9594592365885022, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9594592365885022, |
|
"eval_loss": 0.06401922553777695, |
|
"eval_mean_accuracy": 0.9594592365885022, |
|
"eval_mean_iou": 0.4797296182942511, |
|
"eval_overall_accuracy": 0.9594592365885022, |
|
"eval_runtime": 553.0545, |
|
"eval_samples_per_second": 1.014, |
|
"eval_steps_per_second": 0.128, |
|
"step": 6440 |
|
}, |
|
{ |
|
"epoch": 23.21, |
|
"learning_rate": 2.1065e-05, |
|
"loss": 0.0294, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 23.57, |
|
"learning_rate": 2.0465999999999997e-05, |
|
"loss": 0.0288, |
|
"step": 6600 |
|
}, |
|
{ |
|
"epoch": 23.93, |
|
"learning_rate": 1.9866999999999998e-05, |
|
"loss": 0.0263, |
|
"step": 6700 |
|
}, |
|
{ |
|
"epoch": 24.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9638974238276401, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9638974238276401, |
|
"eval_loss": 0.08045289665460587, |
|
"eval_mean_accuracy": 0.9638974238276401, |
|
"eval_mean_iou": 0.48194871191382005, |
|
"eval_overall_accuracy": 0.9638974238276401, |
|
"eval_runtime": 556.1474, |
|
"eval_samples_per_second": 1.009, |
|
"eval_steps_per_second": 0.128, |
|
"step": 6720 |
|
}, |
|
{ |
|
"epoch": 24.29, |
|
"learning_rate": 1.9267999999999996e-05, |
|
"loss": 0.0253, |
|
"step": 6800 |
|
}, |
|
{ |
|
"epoch": 24.64, |
|
"learning_rate": 1.8669e-05, |
|
"loss": 0.0302, |
|
"step": 6900 |
|
}, |
|
{ |
|
"epoch": 25.0, |
|
"learning_rate": 1.807e-05, |
|
"loss": 0.0252, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 25.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9660654080292466, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9660654080292466, |
|
"eval_loss": 0.07002215832471848, |
|
"eval_mean_accuracy": 0.9660654080292466, |
|
"eval_mean_iou": 0.4830327040146233, |
|
"eval_overall_accuracy": 0.9660654080292466, |
|
"eval_runtime": 543.9929, |
|
"eval_samples_per_second": 1.031, |
|
"eval_steps_per_second": 0.131, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 25.36, |
|
"learning_rate": 1.7471e-05, |
|
"loss": 0.0239, |
|
"step": 7100 |
|
}, |
|
{ |
|
"epoch": 25.71, |
|
"learning_rate": 1.6872e-05, |
|
"loss": 0.0309, |
|
"step": 7200 |
|
}, |
|
{ |
|
"epoch": 26.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9708641099359285, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9708641099359285, |
|
"eval_loss": 0.07471116632223129, |
|
"eval_mean_accuracy": 0.9708641099359285, |
|
"eval_mean_iou": 0.48543205496796427, |
|
"eval_overall_accuracy": 0.9708641099359285, |
|
"eval_runtime": 555.815, |
|
"eval_samples_per_second": 1.009, |
|
"eval_steps_per_second": 0.128, |
|
"step": 7280 |
|
}, |
|
{ |
|
"epoch": 26.07, |
|
"learning_rate": 1.6272999999999998e-05, |
|
"loss": 0.0219, |
|
"step": 7300 |
|
}, |
|
{ |
|
"epoch": 26.43, |
|
"learning_rate": 1.5674e-05, |
|
"loss": 0.0252, |
|
"step": 7400 |
|
}, |
|
{ |
|
"epoch": 26.79, |
|
"learning_rate": 1.5075000000000002e-05, |
|
"loss": 0.0238, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 27.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9627959514469119, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9627959514469119, |
|
"eval_loss": 0.07040922343730927, |
|
"eval_mean_accuracy": 0.9627959514469119, |
|
"eval_mean_iou": 0.48139797572345594, |
|
"eval_overall_accuracy": 0.9627959514469119, |
|
"eval_runtime": 550.1583, |
|
"eval_samples_per_second": 1.02, |
|
"eval_steps_per_second": 0.129, |
|
"step": 7560 |
|
}, |
|
{ |
|
"epoch": 27.14, |
|
"learning_rate": 1.4476e-05, |
|
"loss": 0.0265, |
|
"step": 7600 |
|
}, |
|
{ |
|
"epoch": 27.5, |
|
"learning_rate": 1.3877e-05, |
|
"loss": 0.0248, |
|
"step": 7700 |
|
}, |
|
{ |
|
"epoch": 27.86, |
|
"learning_rate": 1.3277999999999999e-05, |
|
"loss": 0.0277, |
|
"step": 7800 |
|
}, |
|
{ |
|
"epoch": 28.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9715651249947387, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9715651249947387, |
|
"eval_loss": 0.07572752982378006, |
|
"eval_mean_accuracy": 0.9715651249947387, |
|
"eval_mean_iou": 0.48578256249736934, |
|
"eval_overall_accuracy": 0.9715651249947387, |
|
"eval_runtime": 551.1145, |
|
"eval_samples_per_second": 1.018, |
|
"eval_steps_per_second": 0.129, |
|
"step": 7840 |
|
}, |
|
{ |
|
"epoch": 28.21, |
|
"learning_rate": 1.2678999999999998e-05, |
|
"loss": 0.0241, |
|
"step": 7900 |
|
}, |
|
{ |
|
"epoch": 28.57, |
|
"learning_rate": 1.2079999999999998e-05, |
|
"loss": 0.0286, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 28.93, |
|
"learning_rate": 1.1480999999999997e-05, |
|
"loss": 0.0281, |
|
"step": 8100 |
|
}, |
|
{ |
|
"epoch": 29.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9660663529092426, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9660663529092426, |
|
"eval_loss": 0.08465930819511414, |
|
"eval_mean_accuracy": 0.9660663529092426, |
|
"eval_mean_iou": 0.4830331764546213, |
|
"eval_overall_accuracy": 0.9660663529092426, |
|
"eval_runtime": 565.7428, |
|
"eval_samples_per_second": 0.992, |
|
"eval_steps_per_second": 0.125, |
|
"step": 8120 |
|
}, |
|
{ |
|
"epoch": 29.29, |
|
"learning_rate": 1.0882000000000004e-05, |
|
"loss": 0.0249, |
|
"step": 8200 |
|
}, |
|
{ |
|
"epoch": 29.64, |
|
"learning_rate": 1.0283000000000003e-05, |
|
"loss": 0.0256, |
|
"step": 8300 |
|
}, |
|
{ |
|
"epoch": 30.0, |
|
"learning_rate": 9.684000000000002e-06, |
|
"loss": 0.0259, |
|
"step": 8400 |
|
}, |
|
{ |
|
"epoch": 30.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9640005875435611, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9640005875435611, |
|
"eval_loss": 0.0740952268242836, |
|
"eval_mean_accuracy": 0.9640005875435611, |
|
"eval_mean_iou": 0.48200029377178055, |
|
"eval_overall_accuracy": 0.9640005875435611, |
|
"eval_runtime": 555.2353, |
|
"eval_samples_per_second": 1.01, |
|
"eval_steps_per_second": 0.128, |
|
"step": 8400 |
|
}, |
|
{ |
|
"epoch": 30.36, |
|
"learning_rate": 9.085000000000002e-06, |
|
"loss": 0.0229, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 30.71, |
|
"learning_rate": 8.486000000000001e-06, |
|
"loss": 0.0231, |
|
"step": 8600 |
|
}, |
|
{ |
|
"epoch": 31.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9587068544171852, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9587068544171852, |
|
"eval_loss": 0.07262817770242691, |
|
"eval_mean_accuracy": 0.9587068544171852, |
|
"eval_mean_iou": 0.4793534272085926, |
|
"eval_overall_accuracy": 0.9587068544171852, |
|
"eval_runtime": 560.7376, |
|
"eval_samples_per_second": 1.0, |
|
"eval_steps_per_second": 0.127, |
|
"step": 8680 |
|
}, |
|
{ |
|
"epoch": 31.07, |
|
"learning_rate": 7.887000000000001e-06, |
|
"loss": 0.0242, |
|
"step": 8700 |
|
}, |
|
{ |
|
"epoch": 31.43, |
|
"learning_rate": 7.2879999999999995e-06, |
|
"loss": 0.0242, |
|
"step": 8800 |
|
}, |
|
{ |
|
"epoch": 31.79, |
|
"learning_rate": 6.688999999999999e-06, |
|
"loss": 0.0234, |
|
"step": 8900 |
|
}, |
|
{ |
|
"epoch": 32.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.955744140240807, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.955744140240807, |
|
"eval_loss": 0.07388943433761597, |
|
"eval_mean_accuracy": 0.955744140240807, |
|
"eval_mean_iou": 0.4778720701204035, |
|
"eval_overall_accuracy": 0.955744140240807, |
|
"eval_runtime": 574.0863, |
|
"eval_samples_per_second": 0.977, |
|
"eval_steps_per_second": 0.124, |
|
"step": 8960 |
|
}, |
|
{ |
|
"epoch": 32.14, |
|
"learning_rate": 6.0899999999999984e-06, |
|
"loss": 0.0249, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 32.5, |
|
"learning_rate": 5.490999999999998e-06, |
|
"loss": 0.0209, |
|
"step": 9100 |
|
}, |
|
{ |
|
"epoch": 32.86, |
|
"learning_rate": 4.891999999999997e-06, |
|
"loss": 0.0226, |
|
"step": 9200 |
|
}, |
|
{ |
|
"epoch": 33.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9612813088134111, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9612813088134111, |
|
"eval_loss": 0.07430127263069153, |
|
"eval_mean_accuracy": 0.9612813088134111, |
|
"eval_mean_iou": 0.48064065440670556, |
|
"eval_overall_accuracy": 0.9612813088134111, |
|
"eval_runtime": 559.8473, |
|
"eval_samples_per_second": 1.002, |
|
"eval_steps_per_second": 0.127, |
|
"step": 9240 |
|
}, |
|
{ |
|
"epoch": 33.21, |
|
"learning_rate": 4.292999999999997e-06, |
|
"loss": 0.023, |
|
"step": 9300 |
|
}, |
|
{ |
|
"epoch": 33.57, |
|
"learning_rate": 3.694000000000003e-06, |
|
"loss": 0.0226, |
|
"step": 9400 |
|
}, |
|
{ |
|
"epoch": 33.93, |
|
"learning_rate": 3.0950000000000026e-06, |
|
"loss": 0.0242, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 34.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.958420641676595, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.958420641676595, |
|
"eval_loss": 0.07755609601736069, |
|
"eval_mean_accuracy": 0.958420641676595, |
|
"eval_mean_iou": 0.4792103208382975, |
|
"eval_overall_accuracy": 0.958420641676595, |
|
"eval_runtime": 566.607, |
|
"eval_samples_per_second": 0.99, |
|
"eval_steps_per_second": 0.125, |
|
"step": 9520 |
|
}, |
|
{ |
|
"epoch": 34.29, |
|
"learning_rate": 2.496000000000002e-06, |
|
"loss": 0.0224, |
|
"step": 9600 |
|
}, |
|
{ |
|
"epoch": 34.64, |
|
"learning_rate": 1.8970000000000013e-06, |
|
"loss": 0.0233, |
|
"step": 9700 |
|
}, |
|
{ |
|
"epoch": 35.0, |
|
"learning_rate": 1.298000000000001e-06, |
|
"loss": 0.0211, |
|
"step": 9800 |
|
}, |
|
{ |
|
"epoch": 35.0, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9529317478819656, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9529317478819656, |
|
"eval_loss": 0.077478788793087, |
|
"eval_mean_accuracy": 0.9529317478819656, |
|
"eval_mean_iou": 0.4764658739409828, |
|
"eval_overall_accuracy": 0.9529317478819656, |
|
"eval_runtime": 563.7567, |
|
"eval_samples_per_second": 0.995, |
|
"eval_steps_per_second": 0.126, |
|
"step": 9800 |
|
}, |
|
{ |
|
"epoch": 35.36, |
|
"learning_rate": 6.990000000000005e-07, |
|
"loss": 0.0187, |
|
"step": 9900 |
|
}, |
|
{ |
|
"epoch": 35.71, |
|
"learning_rate": 1e-07, |
|
"loss": 0.0223, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 35.71, |
|
"eval_accuracy_background": NaN, |
|
"eval_accuracy_biofilm": 0.9571611166419995, |
|
"eval_iou_background": 0.0, |
|
"eval_iou_biofilm": 0.9571611166419995, |
|
"eval_loss": 0.07972381263971329, |
|
"eval_mean_accuracy": 0.9571611166419995, |
|
"eval_mean_iou": 0.47858055832099977, |
|
"eval_overall_accuracy": 0.9571611166419995, |
|
"eval_runtime": 558.3365, |
|
"eval_samples_per_second": 1.005, |
|
"eval_steps_per_second": 0.127, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 35.71, |
|
"step": 10000, |
|
"total_flos": 1.40223644172288e+18, |
|
"train_loss": 0.037801555216312406, |
|
"train_runtime": 23106.5454, |
|
"train_samples_per_second": 3.462, |
|
"train_steps_per_second": 0.433 |
|
} |
|
], |
|
"logging_steps": 100, |
|
"max_steps": 10000, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 36, |
|
"save_steps": 500, |
|
"total_flos": 1.40223644172288e+18, |
|
"train_batch_size": 8, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|