Koushik Nagasubramanian
End of training
bca5ef1
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 14.749262536873156,
"global_step": 10000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.15,
"learning_rate": 0.0009900010000000001,
"loss": 0.1071,
"step": 100
},
{
"epoch": 0.29,
"learning_rate": 0.000980002,
"loss": 0.05,
"step": 200
},
{
"epoch": 0.44,
"learning_rate": 0.0009700029999999999,
"loss": 0.048,
"step": 300
},
{
"epoch": 0.59,
"learning_rate": 0.0009600039999999999,
"loss": 0.0406,
"step": 400
},
{
"epoch": 0.74,
"learning_rate": 0.0009500049999999999,
"loss": 0.0701,
"step": 500
},
{
"epoch": 0.88,
"learning_rate": 0.0009400059999999998,
"loss": 0.0529,
"step": 600
},
{
"epoch": 1.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 69.37847900390625,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3526.0732,
"eval_samples_per_second": 0.769,
"eval_steps_per_second": 0.048,
"step": 678
},
{
"epoch": 1.03,
"learning_rate": 0.0009300069999999998,
"loss": 0.0387,
"step": 700
},
{
"epoch": 1.18,
"learning_rate": 0.0009200079999999999,
"loss": 0.0361,
"step": 800
},
{
"epoch": 1.33,
"learning_rate": 0.0009100089999999999,
"loss": 0.0355,
"step": 900
},
{
"epoch": 1.47,
"learning_rate": 0.0009000099999999999,
"loss": 0.0414,
"step": 1000
},
{
"epoch": 1.62,
"learning_rate": 0.0008900109999999999,
"loss": 0.0355,
"step": 1100
},
{
"epoch": 1.77,
"learning_rate": 0.0008800119999999999,
"loss": 0.3364,
"step": 1200
},
{
"epoch": 1.92,
"learning_rate": 0.0008700129999999999,
"loss": 0.3755,
"step": 1300
},
{
"epoch": 2.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 0.9454936981201172,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3490.3845,
"eval_samples_per_second": 0.777,
"eval_steps_per_second": 0.049,
"step": 1356
},
{
"epoch": 2.06,
"learning_rate": 0.0008600139999999999,
"loss": 0.0723,
"step": 1400
},
{
"epoch": 2.21,
"learning_rate": 0.0008500149999999999,
"loss": 0.0517,
"step": 1500
},
{
"epoch": 2.36,
"learning_rate": 0.0008400159999999999,
"loss": 0.0537,
"step": 1600
},
{
"epoch": 2.51,
"learning_rate": 0.0008300169999999998,
"loss": 0.06,
"step": 1700
},
{
"epoch": 2.65,
"learning_rate": 0.0008200179999999999,
"loss": 0.0585,
"step": 1800
},
{
"epoch": 2.8,
"learning_rate": 0.0008100189999999999,
"loss": 0.0588,
"step": 1900
},
{
"epoch": 2.95,
"learning_rate": 0.0008000199999999999,
"loss": 0.0603,
"step": 2000
},
{
"epoch": 3.0,
"eval_accuracy_Background": 0.9976430769267056,
"eval_accuracy_Maize": 0.7226873498451214,
"eval_iou_Background": 0.9606768818312337,
"eval_iou_Maize": 0.710616906732821,
"eval_loss": 0.09196273237466812,
"eval_mean_accuracy": 0.8601652133859135,
"eval_mean_iou": 0.8356468942820273,
"eval_overall_accuracy": 0.9641396604486604,
"eval_runtime": 3490.4526,
"eval_samples_per_second": 0.777,
"eval_steps_per_second": 0.049,
"step": 2034
},
{
"epoch": 3.1,
"learning_rate": 0.0007900209999999999,
"loss": 0.0541,
"step": 2100
},
{
"epoch": 3.24,
"learning_rate": 0.0007800219999999999,
"loss": 0.0434,
"step": 2200
},
{
"epoch": 3.39,
"learning_rate": 0.0007700229999999999,
"loss": 0.0407,
"step": 2300
},
{
"epoch": 3.54,
"learning_rate": 0.0007600239999999999,
"loss": 0.0392,
"step": 2400
},
{
"epoch": 3.69,
"learning_rate": 0.000750025,
"loss": 0.0363,
"step": 2500
},
{
"epoch": 3.83,
"learning_rate": 0.000740026,
"loss": 0.0369,
"step": 2600
},
{
"epoch": 3.98,
"learning_rate": 0.000730027,
"loss": 0.0341,
"step": 2700
},
{
"epoch": 4.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 24.620342254638672,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3501.7251,
"eval_samples_per_second": 0.774,
"eval_steps_per_second": 0.049,
"step": 2712
},
{
"epoch": 4.13,
"learning_rate": 0.0007200279999999999,
"loss": 0.0344,
"step": 2800
},
{
"epoch": 4.28,
"learning_rate": 0.0007100289999999999,
"loss": 0.0337,
"step": 2900
},
{
"epoch": 4.42,
"learning_rate": 0.0007000299999999999,
"loss": 0.0327,
"step": 3000
},
{
"epoch": 4.57,
"learning_rate": 0.0006900309999999999,
"loss": 0.0354,
"step": 3100
},
{
"epoch": 4.72,
"learning_rate": 0.0006800319999999999,
"loss": 0.0328,
"step": 3200
},
{
"epoch": 4.87,
"learning_rate": 0.0006700329999999999,
"loss": 0.0332,
"step": 3300
},
{
"epoch": 5.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 101.56349182128906,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3488.5335,
"eval_samples_per_second": 0.777,
"eval_steps_per_second": 0.049,
"step": 3390
},
{
"epoch": 5.01,
"learning_rate": 0.0006600339999999999,
"loss": 0.0328,
"step": 3400
},
{
"epoch": 5.16,
"learning_rate": 0.000650035,
"loss": 0.0326,
"step": 3500
},
{
"epoch": 5.31,
"learning_rate": 0.000640036,
"loss": 0.0311,
"step": 3600
},
{
"epoch": 5.46,
"learning_rate": 0.000630037,
"loss": 0.0348,
"step": 3700
},
{
"epoch": 5.6,
"learning_rate": 0.000620038,
"loss": 0.0318,
"step": 3800
},
{
"epoch": 5.75,
"learning_rate": 0.0006100389999999999,
"loss": 0.0308,
"step": 3900
},
{
"epoch": 5.9,
"learning_rate": 0.0006000399999999999,
"loss": 0.0331,
"step": 4000
},
{
"epoch": 6.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 9.68244743347168,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3519.5902,
"eval_samples_per_second": 0.77,
"eval_steps_per_second": 0.048,
"step": 4068
},
{
"epoch": 6.05,
"learning_rate": 0.000590041,
"loss": 0.0394,
"step": 4100
},
{
"epoch": 6.19,
"learning_rate": 0.0005800420000000001,
"loss": 0.0318,
"step": 4200
},
{
"epoch": 6.34,
"learning_rate": 0.000570043,
"loss": 0.0307,
"step": 4300
},
{
"epoch": 6.49,
"learning_rate": 0.000560044,
"loss": 0.0304,
"step": 4400
},
{
"epoch": 6.64,
"learning_rate": 0.000550045,
"loss": 0.0323,
"step": 4500
},
{
"epoch": 6.78,
"learning_rate": 0.000540046,
"loss": 0.0322,
"step": 4600
},
{
"epoch": 6.93,
"learning_rate": 0.000530047,
"loss": 0.0302,
"step": 4700
},
{
"epoch": 7.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 260.7922668457031,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3539.5206,
"eval_samples_per_second": 0.766,
"eval_steps_per_second": 0.048,
"step": 4746
},
{
"epoch": 7.08,
"learning_rate": 0.000520048,
"loss": 0.0298,
"step": 4800
},
{
"epoch": 7.23,
"learning_rate": 0.0005100489999999999,
"loss": 0.0297,
"step": 4900
},
{
"epoch": 7.37,
"learning_rate": 0.0005000499999999999,
"loss": 0.0336,
"step": 5000
},
{
"epoch": 7.52,
"learning_rate": 0.0004900509999999999,
"loss": 0.0304,
"step": 5100
},
{
"epoch": 7.67,
"learning_rate": 0.00048005199999999996,
"loss": 0.03,
"step": 5200
},
{
"epoch": 7.82,
"learning_rate": 0.00047005299999999995,
"loss": 0.0318,
"step": 5300
},
{
"epoch": 7.96,
"learning_rate": 0.00046005399999999994,
"loss": 0.0305,
"step": 5400
},
{
"epoch": 8.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 172.81529235839844,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3492.8042,
"eval_samples_per_second": 0.776,
"eval_steps_per_second": 0.049,
"step": 5424
},
{
"epoch": 8.11,
"learning_rate": 0.00045005499999999993,
"loss": 0.0294,
"step": 5500
},
{
"epoch": 8.26,
"learning_rate": 0.0004400559999999999,
"loss": 0.0294,
"step": 5600
},
{
"epoch": 8.41,
"learning_rate": 0.000430057,
"loss": 0.0422,
"step": 5700
},
{
"epoch": 8.55,
"learning_rate": 0.000420058,
"loss": 0.0322,
"step": 5800
},
{
"epoch": 8.7,
"learning_rate": 0.000410059,
"loss": 0.03,
"step": 5900
},
{
"epoch": 8.85,
"learning_rate": 0.00040006,
"loss": 0.0366,
"step": 6000
},
{
"epoch": 9.0,
"learning_rate": 0.000390061,
"loss": 0.0313,
"step": 6100
},
{
"epoch": 9.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 304.2713623046875,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3540.1086,
"eval_samples_per_second": 0.766,
"eval_steps_per_second": 0.048,
"step": 6102
},
{
"epoch": 9.14,
"learning_rate": 0.00038006199999999997,
"loss": 0.0314,
"step": 6200
},
{
"epoch": 9.29,
"learning_rate": 0.00037006300000000007,
"loss": 0.03,
"step": 6300
},
{
"epoch": 9.44,
"learning_rate": 0.000360064,
"loss": 0.0296,
"step": 6400
},
{
"epoch": 9.59,
"learning_rate": 0.000350065,
"loss": 0.0298,
"step": 6500
},
{
"epoch": 9.73,
"learning_rate": 0.000340066,
"loss": 0.0298,
"step": 6600
},
{
"epoch": 9.88,
"learning_rate": 0.00033006699999999997,
"loss": 0.0301,
"step": 6700
},
{
"epoch": 10.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 547.2355346679688,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3529.1783,
"eval_samples_per_second": 0.768,
"eval_steps_per_second": 0.048,
"step": 6780
},
{
"epoch": 10.03,
"learning_rate": 0.00032006799999999996,
"loss": 0.0305,
"step": 6800
},
{
"epoch": 10.18,
"learning_rate": 0.00031006900000000006,
"loss": 0.0321,
"step": 6900
},
{
"epoch": 10.32,
"learning_rate": 0.00030007000000000005,
"loss": 0.0294,
"step": 7000
},
{
"epoch": 10.47,
"learning_rate": 0.00029007100000000004,
"loss": 0.0289,
"step": 7100
},
{
"epoch": 10.62,
"learning_rate": 0.000280072,
"loss": 0.0298,
"step": 7200
},
{
"epoch": 10.77,
"learning_rate": 0.000270073,
"loss": 0.0291,
"step": 7300
},
{
"epoch": 10.91,
"learning_rate": 0.000260074,
"loss": 0.03,
"step": 7400
},
{
"epoch": 11.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 224.26065063476562,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3543.4825,
"eval_samples_per_second": 0.765,
"eval_steps_per_second": 0.048,
"step": 7458
},
{
"epoch": 11.06,
"learning_rate": 0.000250075,
"loss": 0.0292,
"step": 7500
},
{
"epoch": 11.21,
"learning_rate": 0.00024007599999999998,
"loss": 0.029,
"step": 7600
},
{
"epoch": 11.36,
"learning_rate": 0.00023007699999999997,
"loss": 0.0283,
"step": 7700
},
{
"epoch": 11.5,
"learning_rate": 0.00022007799999999996,
"loss": 0.0332,
"step": 7800
},
{
"epoch": 11.65,
"learning_rate": 0.00021007899999999995,
"loss": 0.0288,
"step": 7900
},
{
"epoch": 11.8,
"learning_rate": 0.00020007999999999994,
"loss": 0.0286,
"step": 8000
},
{
"epoch": 11.95,
"learning_rate": 0.00019008099999999993,
"loss": 0.0285,
"step": 8100
},
{
"epoch": 12.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 116.34738159179688,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3542.0764,
"eval_samples_per_second": 0.765,
"eval_steps_per_second": 0.048,
"step": 8136
},
{
"epoch": 12.09,
"learning_rate": 0.00018008200000000006,
"loss": 0.0284,
"step": 8200
},
{
"epoch": 12.24,
"learning_rate": 0.00017008300000000005,
"loss": 0.0286,
"step": 8300
},
{
"epoch": 12.39,
"learning_rate": 0.00016008400000000003,
"loss": 0.028,
"step": 8400
},
{
"epoch": 12.54,
"learning_rate": 0.00015008500000000002,
"loss": 0.0312,
"step": 8500
},
{
"epoch": 12.68,
"learning_rate": 0.00014008600000000001,
"loss": 0.0283,
"step": 8600
},
{
"epoch": 12.83,
"learning_rate": 0.000130087,
"loss": 0.0287,
"step": 8700
},
{
"epoch": 12.98,
"learning_rate": 0.00012008799999999999,
"loss": 0.0284,
"step": 8800
},
{
"epoch": 13.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 96.84294128417969,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3520.8181,
"eval_samples_per_second": 0.77,
"eval_steps_per_second": 0.048,
"step": 8814
},
{
"epoch": 13.13,
"learning_rate": 0.00011008899999999998,
"loss": 0.0281,
"step": 8900
},
{
"epoch": 13.27,
"learning_rate": 0.00010008999999999997,
"loss": 0.0287,
"step": 9000
},
{
"epoch": 13.42,
"learning_rate": 9.009099999999997e-05,
"loss": 0.0284,
"step": 9100
},
{
"epoch": 13.57,
"learning_rate": 8.009199999999996e-05,
"loss": 0.0287,
"step": 9200
},
{
"epoch": 13.72,
"learning_rate": 7.009299999999995e-05,
"loss": 0.0284,
"step": 9300
},
{
"epoch": 13.86,
"learning_rate": 6.009400000000005e-05,
"loss": 0.0281,
"step": 9400
},
{
"epoch": 14.0,
"eval_accuracy_Background": 1.0,
"eval_accuracy_Maize": 0.0,
"eval_iou_Background": 0.8781497776618262,
"eval_iou_Maize": 0.0,
"eval_loss": 54.25930404663086,
"eval_mean_accuracy": 0.5,
"eval_mean_iou": 0.4390748888309131,
"eval_overall_accuracy": 0.8781497776618262,
"eval_runtime": 3530.991,
"eval_samples_per_second": 0.768,
"eval_steps_per_second": 0.048,
"step": 9492
},
{
"epoch": 14.01,
"learning_rate": 5.009500000000005e-05,
"loss": 0.028,
"step": 9500
},
{
"epoch": 14.16,
"learning_rate": 4.009600000000004e-05,
"loss": 0.0283,
"step": 9600
},
{
"epoch": 14.31,
"learning_rate": 3.0097000000000024e-05,
"loss": 0.0279,
"step": 9700
},
{
"epoch": 14.45,
"learning_rate": 2.0098000000000017e-05,
"loss": 0.0282,
"step": 9800
},
{
"epoch": 14.6,
"learning_rate": 1.009900000000001e-05,
"loss": 0.028,
"step": 9900
},
{
"epoch": 14.75,
"learning_rate": 1e-07,
"loss": 0.028,
"step": 10000
},
{
"epoch": 14.75,
"eval_accuracy_Background": 0.9834420618324031,
"eval_accuracy_Maize": 0.9587911337930981,
"eval_iou_Background": 0.9778506571216238,
"eval_iou_Maize": 0.8565761625984988,
"eval_loss": 0.07561617344617844,
"eval_mean_accuracy": 0.9711165978127506,
"eval_mean_iou": 0.9172134098600613,
"eval_overall_accuracy": 0.9804383407699715,
"eval_runtime": 3519.0292,
"eval_samples_per_second": 0.77,
"eval_steps_per_second": 0.048,
"step": 10000
},
{
"epoch": 14.75,
"step": 10000,
"total_flos": 6.383905554004181e+19,
"train_loss": 0.042073391890525814,
"train_runtime": 74840.5696,
"train_samples_per_second": 2.138,
"train_steps_per_second": 0.134
}
],
"max_steps": 10000,
"num_train_epochs": 15,
"total_flos": 6.383905554004181e+19,
"trial_name": null,
"trial_params": null
}