sam1120's picture
Training in progress, step 120
ad44638
raw
history blame
21.9 kB
{
"best_metric": 0.31272849440574646,
"best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGBD-b5_5/checkpoint-120",
"epoch": 60.0,
"global_step": 120,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.5,
"learning_rate": 7.5e-07,
"loss": 1.2301,
"step": 1
},
{
"epoch": 1.0,
"learning_rate": 1.5e-06,
"loss": 1.2312,
"step": 2
},
{
"epoch": 1.5,
"learning_rate": 2.25e-06,
"loss": 1.2262,
"step": 3
},
{
"epoch": 2.0,
"learning_rate": 3e-06,
"loss": 1.2197,
"step": 4
},
{
"epoch": 2.5,
"learning_rate": 3.75e-06,
"loss": 1.2077,
"step": 5
},
{
"epoch": 3.0,
"learning_rate": 4.5e-06,
"loss": 1.1967,
"step": 6
},
{
"epoch": 3.5,
"learning_rate": 5.2500000000000006e-06,
"loss": 1.1848,
"step": 7
},
{
"epoch": 4.0,
"learning_rate": 6e-06,
"loss": 1.1668,
"step": 8
},
{
"epoch": 4.5,
"learning_rate": 6.75e-06,
"loss": 1.1606,
"step": 9
},
{
"epoch": 5.0,
"learning_rate": 7.5e-06,
"loss": 1.1487,
"step": 10
},
{
"epoch": 5.0,
"eval_accuracy_dropoff": 0.5729856189362141,
"eval_accuracy_undropoff": 0.6823051671829305,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.09713754230539519,
"eval_iou_undropoff": 0.671356002033353,
"eval_iou_unlabeled": 0.0,
"eval_loss": 1.0250014066696167,
"eval_mean_accuracy": 0.6276453930595723,
"eval_mean_iou": 0.25616451477958274,
"eval_overall_accuracy": 0.6777597427368164,
"eval_runtime": 2.8634,
"eval_samples_per_second": 6.985,
"eval_steps_per_second": 0.698,
"step": 10
},
{
"epoch": 5.5,
"learning_rate": 8.25e-06,
"loss": 1.1397,
"step": 11
},
{
"epoch": 6.0,
"learning_rate": 9e-06,
"loss": 1.1186,
"step": 12
},
{
"epoch": 6.5,
"learning_rate": 8.960526315789474e-06,
"loss": 1.1081,
"step": 13
},
{
"epoch": 7.0,
"learning_rate": 8.921052631578947e-06,
"loss": 1.0962,
"step": 14
},
{
"epoch": 7.5,
"learning_rate": 8.881578947368421e-06,
"loss": 1.0719,
"step": 15
},
{
"epoch": 8.0,
"learning_rate": 8.842105263157895e-06,
"loss": 1.0773,
"step": 16
},
{
"epoch": 8.5,
"learning_rate": 8.80263157894737e-06,
"loss": 1.0485,
"step": 17
},
{
"epoch": 9.0,
"learning_rate": 8.763157894736842e-06,
"loss": 1.0373,
"step": 18
},
{
"epoch": 9.5,
"learning_rate": 8.723684210526316e-06,
"loss": 1.0212,
"step": 19
},
{
"epoch": 10.0,
"learning_rate": 8.68421052631579e-06,
"loss": 1.0128,
"step": 20
},
{
"epoch": 10.0,
"eval_accuracy_dropoff": 0.5053189293332416,
"eval_accuracy_undropoff": 0.8407032200736932,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.11952306654022513,
"eval_iou_undropoff": 0.8230702761890567,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.9030143022537231,
"eval_mean_accuracy": 0.6730110747034674,
"eval_mean_iou": 0.31419778090976064,
"eval_overall_accuracy": 0.8267581939697266,
"eval_runtime": 2.9046,
"eval_samples_per_second": 6.886,
"eval_steps_per_second": 0.689,
"step": 20
},
{
"epoch": 10.5,
"learning_rate": 8.644736842105263e-06,
"loss": 0.9768,
"step": 21
},
{
"epoch": 11.0,
"learning_rate": 8.605263157894737e-06,
"loss": 0.992,
"step": 22
},
{
"epoch": 11.5,
"learning_rate": 8.56578947368421e-06,
"loss": 0.9632,
"step": 23
},
{
"epoch": 12.0,
"learning_rate": 8.526315789473683e-06,
"loss": 0.9467,
"step": 24
},
{
"epoch": 12.5,
"learning_rate": 8.486842105263159e-06,
"loss": 0.9267,
"step": 25
},
{
"epoch": 13.0,
"learning_rate": 8.447368421052632e-06,
"loss": 0.9105,
"step": 26
},
{
"epoch": 13.5,
"learning_rate": 8.407894736842106e-06,
"loss": 0.8971,
"step": 27
},
{
"epoch": 14.0,
"learning_rate": 8.36842105263158e-06,
"loss": 0.8905,
"step": 28
},
{
"epoch": 14.5,
"learning_rate": 8.328947368421052e-06,
"loss": 0.8751,
"step": 29
},
{
"epoch": 15.0,
"learning_rate": 8.289473684210526e-06,
"loss": 0.8561,
"step": 30
},
{
"epoch": 15.0,
"eval_accuracy_dropoff": 0.4692355329250671,
"eval_accuracy_undropoff": 0.9133466736054656,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.1631859638151545,
"eval_iou_undropoff": 0.8928195277656624,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.7358734607696533,
"eval_mean_accuracy": 0.6912911032652663,
"eval_mean_iou": 0.35200183052693895,
"eval_overall_accuracy": 0.8948808670043945,
"eval_runtime": 9.5774,
"eval_samples_per_second": 2.088,
"eval_steps_per_second": 0.209,
"step": 30
},
{
"epoch": 15.5,
"learning_rate": 8.25e-06,
"loss": 0.8539,
"step": 31
},
{
"epoch": 16.0,
"learning_rate": 8.210526315789473e-06,
"loss": 0.8166,
"step": 32
},
{
"epoch": 16.5,
"learning_rate": 8.171052631578947e-06,
"loss": 0.8398,
"step": 33
},
{
"epoch": 17.0,
"learning_rate": 8.131578947368421e-06,
"loss": 0.7988,
"step": 34
},
{
"epoch": 17.5,
"learning_rate": 8.092105263157896e-06,
"loss": 0.7788,
"step": 35
},
{
"epoch": 18.0,
"learning_rate": 8.052631578947368e-06,
"loss": 0.7842,
"step": 36
},
{
"epoch": 18.5,
"learning_rate": 8.013157894736842e-06,
"loss": 0.7554,
"step": 37
},
{
"epoch": 19.0,
"learning_rate": 7.973684210526316e-06,
"loss": 0.746,
"step": 38
},
{
"epoch": 19.5,
"learning_rate": 7.934210526315789e-06,
"loss": 0.7234,
"step": 39
},
{
"epoch": 20.0,
"learning_rate": 7.894736842105263e-06,
"loss": 0.7551,
"step": 40
},
{
"epoch": 20.0,
"eval_accuracy_dropoff": 0.47188696988463036,
"eval_accuracy_undropoff": 0.9279577542570626,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.18289512986089382,
"eval_iou_undropoff": 0.9072130629563177,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.6533988118171692,
"eval_mean_accuracy": 0.6999223620708465,
"eval_mean_iou": 0.3633693976057371,
"eval_overall_accuracy": 0.9089946746826172,
"eval_runtime": 3.4912,
"eval_samples_per_second": 5.729,
"eval_steps_per_second": 0.573,
"step": 40
},
{
"epoch": 20.5,
"learning_rate": 7.855263157894737e-06,
"loss": 0.6963,
"step": 41
},
{
"epoch": 21.0,
"learning_rate": 7.815789473684211e-06,
"loss": 0.6967,
"step": 42
},
{
"epoch": 21.5,
"learning_rate": 7.776315789473685e-06,
"loss": 0.6664,
"step": 43
},
{
"epoch": 22.0,
"learning_rate": 7.736842105263158e-06,
"loss": 0.6761,
"step": 44
},
{
"epoch": 22.5,
"learning_rate": 7.697368421052632e-06,
"loss": 0.6465,
"step": 45
},
{
"epoch": 23.0,
"learning_rate": 7.657894736842106e-06,
"loss": 0.6876,
"step": 46
},
{
"epoch": 23.5,
"learning_rate": 7.618421052631579e-06,
"loss": 0.6282,
"step": 47
},
{
"epoch": 24.0,
"learning_rate": 7.578947368421053e-06,
"loss": 0.6247,
"step": 48
},
{
"epoch": 24.5,
"learning_rate": 7.539473684210527e-06,
"loss": 0.6037,
"step": 49
},
{
"epoch": 25.0,
"learning_rate": 7.5e-06,
"loss": 0.6236,
"step": 50
},
{
"epoch": 25.0,
"eval_accuracy_dropoff": 0.4614004908369458,
"eval_accuracy_undropoff": 0.9387625786460785,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.19552168013778248,
"eval_iou_undropoff": 0.9173372880255048,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.5938196778297424,
"eval_mean_accuracy": 0.7000815347415121,
"eval_mean_iou": 0.3709529893877625,
"eval_overall_accuracy": 0.9189142227172852,
"eval_runtime": 2.999,
"eval_samples_per_second": 6.669,
"eval_steps_per_second": 0.667,
"step": 50
},
{
"epoch": 25.5,
"learning_rate": 7.460526315789474e-06,
"loss": 0.5851,
"step": 51
},
{
"epoch": 26.0,
"learning_rate": 7.421052631578948e-06,
"loss": 0.5998,
"step": 52
},
{
"epoch": 26.5,
"learning_rate": 7.381578947368421e-06,
"loss": 0.5867,
"step": 53
},
{
"epoch": 27.0,
"learning_rate": 7.342105263157895e-06,
"loss": 0.5478,
"step": 54
},
{
"epoch": 27.5,
"learning_rate": 7.302631578947368e-06,
"loss": 0.5257,
"step": 55
},
{
"epoch": 28.0,
"learning_rate": 7.263157894736842e-06,
"loss": 0.5565,
"step": 56
},
{
"epoch": 28.5,
"learning_rate": 7.223684210526317e-06,
"loss": 0.5253,
"step": 57
},
{
"epoch": 29.0,
"learning_rate": 7.18421052631579e-06,
"loss": 0.5249,
"step": 58
},
{
"epoch": 29.5,
"learning_rate": 7.144736842105263e-06,
"loss": 0.5155,
"step": 59
},
{
"epoch": 30.0,
"learning_rate": 7.1052631578947375e-06,
"loss": 0.4977,
"step": 60
},
{
"epoch": 30.0,
"eval_accuracy_dropoff": 0.44204683593660404,
"eval_accuracy_undropoff": 0.9554527118531071,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.22222683445987237,
"eval_iou_undropoff": 0.9328776694924146,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.5293418169021606,
"eval_mean_accuracy": 0.6987497738948556,
"eval_mean_iou": 0.3850348346507623,
"eval_overall_accuracy": 0.9341056823730469,
"eval_runtime": 2.9259,
"eval_samples_per_second": 6.836,
"eval_steps_per_second": 0.684,
"step": 60
},
{
"epoch": 30.5,
"learning_rate": 7.065789473684211e-06,
"loss": 0.486,
"step": 61
},
{
"epoch": 31.0,
"learning_rate": 7.026315789473684e-06,
"loss": 0.4739,
"step": 62
},
{
"epoch": 31.5,
"learning_rate": 6.986842105263158e-06,
"loss": 0.465,
"step": 63
},
{
"epoch": 32.0,
"learning_rate": 6.9473684210526315e-06,
"loss": 0.4841,
"step": 64
},
{
"epoch": 32.5,
"learning_rate": 6.907894736842106e-06,
"loss": 0.4661,
"step": 65
},
{
"epoch": 33.0,
"learning_rate": 6.86842105263158e-06,
"loss": 0.457,
"step": 66
},
{
"epoch": 33.5,
"learning_rate": 6.828947368421053e-06,
"loss": 0.4309,
"step": 67
},
{
"epoch": 34.0,
"learning_rate": 6.7894736842105264e-06,
"loss": 0.4359,
"step": 68
},
{
"epoch": 34.5,
"learning_rate": 6.75e-06,
"loss": 0.414,
"step": 69
},
{
"epoch": 35.0,
"learning_rate": 6.710526315789474e-06,
"loss": 0.4188,
"step": 70
},
{
"epoch": 35.0,
"eval_accuracy_dropoff": 0.42312897084795525,
"eval_accuracy_undropoff": 0.9650274583398426,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.23898106598406102,
"eval_iou_undropoff": 0.9414664632743484,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.48594316840171814,
"eval_mean_accuracy": 0.694078214593899,
"eval_mean_iou": 0.39348250975280313,
"eval_overall_accuracy": 0.9424957275390625,
"eval_runtime": 3.0215,
"eval_samples_per_second": 6.619,
"eval_steps_per_second": 0.662,
"step": 70
},
{
"epoch": 35.5,
"learning_rate": 6.671052631578947e-06,
"loss": 0.397,
"step": 71
},
{
"epoch": 36.0,
"learning_rate": 6.6315789473684205e-06,
"loss": 0.4114,
"step": 72
},
{
"epoch": 36.5,
"learning_rate": 6.592105263157895e-06,
"loss": 0.4052,
"step": 73
},
{
"epoch": 37.0,
"learning_rate": 6.552631578947369e-06,
"loss": 0.404,
"step": 74
},
{
"epoch": 37.5,
"learning_rate": 6.513157894736842e-06,
"loss": 0.3704,
"step": 75
},
{
"epoch": 38.0,
"learning_rate": 6.473684210526316e-06,
"loss": 0.4008,
"step": 76
},
{
"epoch": 38.5,
"learning_rate": 6.4342105263157896e-06,
"loss": 0.3855,
"step": 77
},
{
"epoch": 39.0,
"learning_rate": 6.394736842105263e-06,
"loss": 0.3719,
"step": 78
},
{
"epoch": 39.5,
"learning_rate": 6.355263157894737e-06,
"loss": 0.3637,
"step": 79
},
{
"epoch": 40.0,
"learning_rate": 6.31578947368421e-06,
"loss": 0.3532,
"step": 80
},
{
"epoch": 40.0,
"eval_accuracy_dropoff": 0.38814651712195236,
"eval_accuracy_undropoff": 0.9763837381353007,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.25466661048474964,
"eval_iou_undropoff": 0.9511414746114953,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.42775267362594604,
"eval_mean_accuracy": 0.6822651276286265,
"eval_mean_iou": 0.401936028365415,
"eval_overall_accuracy": 0.9519252777099609,
"eval_runtime": 2.9058,
"eval_samples_per_second": 6.883,
"eval_steps_per_second": 0.688,
"step": 80
},
{
"epoch": 40.5,
"learning_rate": 6.276315789473684e-06,
"loss": 0.343,
"step": 81
},
{
"epoch": 41.0,
"learning_rate": 6.236842105263159e-06,
"loss": 0.3551,
"step": 82
},
{
"epoch": 41.5,
"learning_rate": 6.197368421052632e-06,
"loss": 0.3465,
"step": 83
},
{
"epoch": 42.0,
"learning_rate": 6.157894736842105e-06,
"loss": 0.3327,
"step": 84
},
{
"epoch": 42.5,
"learning_rate": 6.118421052631579e-06,
"loss": 0.3481,
"step": 85
},
{
"epoch": 43.0,
"learning_rate": 6.078947368421053e-06,
"loss": 0.3134,
"step": 86
},
{
"epoch": 43.5,
"learning_rate": 6.039473684210526e-06,
"loss": 0.3053,
"step": 87
},
{
"epoch": 44.0,
"learning_rate": 6e-06,
"loss": 0.32,
"step": 88
},
{
"epoch": 44.5,
"learning_rate": 5.9605263157894735e-06,
"loss": 0.3236,
"step": 89
},
{
"epoch": 45.0,
"learning_rate": 5.921052631578948e-06,
"loss": 0.3187,
"step": 90
},
{
"epoch": 45.0,
"eval_accuracy_dropoff": 0.39418794009036906,
"eval_accuracy_undropoff": 0.9803864168035686,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.2742061579993682,
"eval_iou_undropoff": 0.9552955051752383,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.3913882076740265,
"eval_mean_accuracy": 0.6872871784469688,
"eval_mean_iou": 0.40983388772486884,
"eval_overall_accuracy": 0.9560127258300781,
"eval_runtime": 3.0009,
"eval_samples_per_second": 6.665,
"eval_steps_per_second": 0.666,
"step": 90
},
{
"epoch": 45.5,
"learning_rate": 5.881578947368422e-06,
"loss": 0.3086,
"step": 91
},
{
"epoch": 46.0,
"learning_rate": 5.842105263157895e-06,
"loss": 0.2949,
"step": 92
},
{
"epoch": 46.5,
"learning_rate": 5.802631578947368e-06,
"loss": 0.3289,
"step": 93
},
{
"epoch": 47.0,
"learning_rate": 5.7631578947368425e-06,
"loss": 0.2915,
"step": 94
},
{
"epoch": 47.5,
"learning_rate": 5.723684210526316e-06,
"loss": 0.29,
"step": 95
},
{
"epoch": 48.0,
"learning_rate": 5.684210526315789e-06,
"loss": 0.3007,
"step": 96
},
{
"epoch": 48.5,
"learning_rate": 5.644736842105263e-06,
"loss": 0.3019,
"step": 97
},
{
"epoch": 49.0,
"learning_rate": 5.6052631578947374e-06,
"loss": 0.2885,
"step": 98
},
{
"epoch": 49.5,
"learning_rate": 5.565789473684211e-06,
"loss": 0.2814,
"step": 99
},
{
"epoch": 50.0,
"learning_rate": 5.526315789473685e-06,
"loss": 0.2631,
"step": 100
},
{
"epoch": 50.0,
"eval_accuracy_dropoff": 0.4020138076561389,
"eval_accuracy_undropoff": 0.9815484334467356,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.28354789968712896,
"eval_iou_undropoff": 0.9567292612427862,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.3646724224090576,
"eval_mean_accuracy": 0.6917811205514373,
"eval_mean_iou": 0.41342572030997177,
"eval_overall_accuracy": 0.9574518203735352,
"eval_runtime": 2.934,
"eval_samples_per_second": 6.817,
"eval_steps_per_second": 0.682,
"step": 100
},
{
"epoch": 50.5,
"learning_rate": 5.486842105263158e-06,
"loss": 0.2683,
"step": 101
},
{
"epoch": 51.0,
"learning_rate": 5.4473684210526315e-06,
"loss": 0.2636,
"step": 102
},
{
"epoch": 51.5,
"learning_rate": 5.407894736842106e-06,
"loss": 0.2767,
"step": 103
},
{
"epoch": 52.0,
"learning_rate": 5.368421052631579e-06,
"loss": 0.2579,
"step": 104
},
{
"epoch": 52.5,
"learning_rate": 5.328947368421052e-06,
"loss": 0.2612,
"step": 105
},
{
"epoch": 53.0,
"learning_rate": 5.289473684210527e-06,
"loss": 0.2473,
"step": 106
},
{
"epoch": 53.5,
"learning_rate": 5.2500000000000006e-06,
"loss": 0.2459,
"step": 107
},
{
"epoch": 54.0,
"learning_rate": 5.210526315789474e-06,
"loss": 0.2658,
"step": 108
},
{
"epoch": 54.5,
"learning_rate": 5.171052631578947e-06,
"loss": 0.2496,
"step": 109
},
{
"epoch": 55.0,
"learning_rate": 5.131578947368421e-06,
"loss": 0.2565,
"step": 110
},
{
"epoch": 55.0,
"eval_accuracy_dropoff": 0.3962109222688594,
"eval_accuracy_undropoff": 0.9828545329893122,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.2845630508195371,
"eval_iou_undropoff": 0.957769917133554,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.34244000911712646,
"eval_mean_accuracy": 0.6895327276290858,
"eval_mean_iou": 0.414110989317697,
"eval_overall_accuracy": 0.9584623336791992,
"eval_runtime": 2.879,
"eval_samples_per_second": 6.947,
"eval_steps_per_second": 0.695,
"step": 110
},
{
"epoch": 55.5,
"learning_rate": 5.092105263157895e-06,
"loss": 0.2835,
"step": 111
},
{
"epoch": 56.0,
"learning_rate": 5.052631578947368e-06,
"loss": 0.2464,
"step": 112
},
{
"epoch": 56.5,
"learning_rate": 5.013157894736842e-06,
"loss": 0.24,
"step": 113
},
{
"epoch": 57.0,
"learning_rate": 4.973684210526316e-06,
"loss": 0.2329,
"step": 114
},
{
"epoch": 57.5,
"learning_rate": 4.9342105263157895e-06,
"loss": 0.2399,
"step": 115
},
{
"epoch": 58.0,
"learning_rate": 4.894736842105264e-06,
"loss": 0.2468,
"step": 116
},
{
"epoch": 58.5,
"learning_rate": 4.855263157894737e-06,
"loss": 0.2338,
"step": 117
},
{
"epoch": 59.0,
"learning_rate": 4.81578947368421e-06,
"loss": 0.228,
"step": 118
},
{
"epoch": 59.5,
"learning_rate": 4.7763157894736844e-06,
"loss": 0.2336,
"step": 119
},
{
"epoch": 60.0,
"learning_rate": 4.736842105263158e-06,
"loss": 0.2259,
"step": 120
},
{
"epoch": 60.0,
"eval_accuracy_dropoff": 0.3842794559508246,
"eval_accuracy_undropoff": 0.9863174978133828,
"eval_accuracy_unlabeled": NaN,
"eval_iou_dropoff": 0.2925986727209221,
"eval_iou_undropoff": 0.9606573233147998,
"eval_iou_unlabeled": 0.0,
"eval_loss": 0.31272849440574646,
"eval_mean_accuracy": 0.6852984768821038,
"eval_mean_iou": 0.417751998678574,
"eval_overall_accuracy": 0.9612852096557617,
"eval_runtime": 14.3976,
"eval_samples_per_second": 1.389,
"eval_steps_per_second": 0.139,
"step": 120
}
],
"max_steps": 240,
"num_train_epochs": 120,
"total_flos": 1.3892743058581094e+18,
"trial_name": null,
"trial_params": null
}