|
{ |
|
"best_metric": 0.7174801826477051, |
|
"best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/parking-utcustom-train-SF-RGB-b0_6/checkpoint-80", |
|
"epoch": 80.0, |
|
"global_step": 80, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 7.5e-06, |
|
"loss": 1.2634, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 1.5e-05, |
|
"loss": 1.2594, |
|
"step": 2 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"learning_rate": 2.25e-05, |
|
"loss": 1.2595, |
|
"step": 3 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"learning_rate": 3e-05, |
|
"loss": 1.2555, |
|
"step": 4 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"learning_rate": 3.7500000000000003e-05, |
|
"loss": 1.2421, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 6.0, |
|
"learning_rate": 4.5e-05, |
|
"loss": 1.2367, |
|
"step": 6 |
|
}, |
|
{ |
|
"epoch": 7.0, |
|
"learning_rate": 4.460526315789474e-05, |
|
"loss": 1.2229, |
|
"step": 7 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"learning_rate": 4.421052631578947e-05, |
|
"loss": 1.2184, |
|
"step": 8 |
|
}, |
|
{ |
|
"epoch": 9.0, |
|
"learning_rate": 4.3815789473684214e-05, |
|
"loss": 1.1978, |
|
"step": 9 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"learning_rate": 4.342105263157895e-05, |
|
"loss": 1.1903, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 11.0, |
|
"learning_rate": 4.302631578947368e-05, |
|
"loss": 1.1701, |
|
"step": 11 |
|
}, |
|
{ |
|
"epoch": 12.0, |
|
"learning_rate": 4.263157894736842e-05, |
|
"loss": 1.1702, |
|
"step": 12 |
|
}, |
|
{ |
|
"epoch": 13.0, |
|
"learning_rate": 4.223684210526316e-05, |
|
"loss": 1.1505, |
|
"step": 13 |
|
}, |
|
{ |
|
"epoch": 14.0, |
|
"learning_rate": 4.1842105263157894e-05, |
|
"loss": 1.1512, |
|
"step": 14 |
|
}, |
|
{ |
|
"epoch": 15.0, |
|
"learning_rate": 4.1447368421052636e-05, |
|
"loss": 1.125, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 16.0, |
|
"learning_rate": 4.105263157894737e-05, |
|
"loss": 1.1275, |
|
"step": 16 |
|
}, |
|
{ |
|
"epoch": 17.0, |
|
"learning_rate": 4.065789473684211e-05, |
|
"loss": 1.115, |
|
"step": 17 |
|
}, |
|
{ |
|
"epoch": 18.0, |
|
"learning_rate": 4.026315789473685e-05, |
|
"loss": 1.1135, |
|
"step": 18 |
|
}, |
|
{ |
|
"epoch": 19.0, |
|
"learning_rate": 3.986842105263158e-05, |
|
"loss": 1.0903, |
|
"step": 19 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"learning_rate": 3.9473684210526316e-05, |
|
"loss": 1.0679, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"eval_accuracy_parking": NaN, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_accuracy_unparking": 0.6371002197265625, |
|
"eval_iou_parking": 0.0, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_iou_unparking": 0.6371002197265625, |
|
"eval_loss": 1.084120273590088, |
|
"eval_mean_accuracy": 0.6371002197265625, |
|
"eval_mean_iou": 0.21236673990885416, |
|
"eval_overall_accuracy": 0.6371002197265625, |
|
"eval_runtime": 0.481, |
|
"eval_samples_per_second": 2.079, |
|
"eval_steps_per_second": 2.079, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 21.0, |
|
"learning_rate": 3.907894736842106e-05, |
|
"loss": 1.0759, |
|
"step": 21 |
|
}, |
|
{ |
|
"epoch": 22.0, |
|
"learning_rate": 3.868421052631579e-05, |
|
"loss": 1.0536, |
|
"step": 22 |
|
}, |
|
{ |
|
"epoch": 23.0, |
|
"learning_rate": 3.828947368421053e-05, |
|
"loss": 1.0421, |
|
"step": 23 |
|
}, |
|
{ |
|
"epoch": 24.0, |
|
"learning_rate": 3.789473684210526e-05, |
|
"loss": 1.0189, |
|
"step": 24 |
|
}, |
|
{ |
|
"epoch": 25.0, |
|
"learning_rate": 3.7500000000000003e-05, |
|
"loss": 1.0057, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 26.0, |
|
"learning_rate": 3.710526315789474e-05, |
|
"loss": 0.9971, |
|
"step": 26 |
|
}, |
|
{ |
|
"epoch": 27.0, |
|
"learning_rate": 3.671052631578947e-05, |
|
"loss": 1.0254, |
|
"step": 27 |
|
}, |
|
{ |
|
"epoch": 28.0, |
|
"learning_rate": 3.631578947368421e-05, |
|
"loss": 0.9999, |
|
"step": 28 |
|
}, |
|
{ |
|
"epoch": 29.0, |
|
"learning_rate": 3.592105263157895e-05, |
|
"loss": 1.008, |
|
"step": 29 |
|
}, |
|
{ |
|
"epoch": 30.0, |
|
"learning_rate": 3.5526315789473684e-05, |
|
"loss": 0.9677, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 31.0, |
|
"learning_rate": 3.5131578947368425e-05, |
|
"loss": 0.9522, |
|
"step": 31 |
|
}, |
|
{ |
|
"epoch": 32.0, |
|
"learning_rate": 3.473684210526316e-05, |
|
"loss": 0.9452, |
|
"step": 32 |
|
}, |
|
{ |
|
"epoch": 33.0, |
|
"learning_rate": 3.43421052631579e-05, |
|
"loss": 0.9289, |
|
"step": 33 |
|
}, |
|
{ |
|
"epoch": 34.0, |
|
"learning_rate": 3.3947368421052636e-05, |
|
"loss": 0.9161, |
|
"step": 34 |
|
}, |
|
{ |
|
"epoch": 35.0, |
|
"learning_rate": 3.355263157894737e-05, |
|
"loss": 0.9317, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 36.0, |
|
"learning_rate": 3.3157894736842106e-05, |
|
"loss": 0.908, |
|
"step": 36 |
|
}, |
|
{ |
|
"epoch": 37.0, |
|
"learning_rate": 3.276315789473685e-05, |
|
"loss": 0.8924, |
|
"step": 37 |
|
}, |
|
{ |
|
"epoch": 38.0, |
|
"learning_rate": 3.236842105263158e-05, |
|
"loss": 0.8983, |
|
"step": 38 |
|
}, |
|
{ |
|
"epoch": 39.0, |
|
"learning_rate": 3.197368421052632e-05, |
|
"loss": 0.8536, |
|
"step": 39 |
|
}, |
|
{ |
|
"epoch": 40.0, |
|
"learning_rate": 3.157894736842105e-05, |
|
"loss": 0.8545, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 40.0, |
|
"eval_accuracy_parking": NaN, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_accuracy_unparking": 0.8015785217285156, |
|
"eval_iou_parking": 0.0, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_iou_unparking": 0.8015785217285156, |
|
"eval_loss": 1.0206384658813477, |
|
"eval_mean_accuracy": 0.8015785217285156, |
|
"eval_mean_iou": 0.2671928405761719, |
|
"eval_overall_accuracy": 0.8015785217285156, |
|
"eval_runtime": 0.4816, |
|
"eval_samples_per_second": 2.076, |
|
"eval_steps_per_second": 2.076, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 41.0, |
|
"learning_rate": 3.118421052631579e-05, |
|
"loss": 0.8371, |
|
"step": 41 |
|
}, |
|
{ |
|
"epoch": 42.0, |
|
"learning_rate": 3.078947368421053e-05, |
|
"loss": 0.8409, |
|
"step": 42 |
|
}, |
|
{ |
|
"epoch": 43.0, |
|
"learning_rate": 3.0394736842105266e-05, |
|
"loss": 0.8354, |
|
"step": 43 |
|
}, |
|
{ |
|
"epoch": 44.0, |
|
"learning_rate": 3e-05, |
|
"loss": 0.8029, |
|
"step": 44 |
|
}, |
|
{ |
|
"epoch": 45.0, |
|
"learning_rate": 2.9605263157894742e-05, |
|
"loss": 0.8197, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 46.0, |
|
"learning_rate": 2.9210526315789477e-05, |
|
"loss": 0.8046, |
|
"step": 46 |
|
}, |
|
{ |
|
"epoch": 47.0, |
|
"learning_rate": 2.8815789473684212e-05, |
|
"loss": 0.7741, |
|
"step": 47 |
|
}, |
|
{ |
|
"epoch": 48.0, |
|
"learning_rate": 2.8421052631578946e-05, |
|
"loss": 0.7917, |
|
"step": 48 |
|
}, |
|
{ |
|
"epoch": 49.0, |
|
"learning_rate": 2.8026315789473688e-05, |
|
"loss": 0.7832, |
|
"step": 49 |
|
}, |
|
{ |
|
"epoch": 50.0, |
|
"learning_rate": 2.7631578947368423e-05, |
|
"loss": 0.7764, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 51.0, |
|
"learning_rate": 2.723684210526316e-05, |
|
"loss": 0.7553, |
|
"step": 51 |
|
}, |
|
{ |
|
"epoch": 52.0, |
|
"learning_rate": 2.6842105263157896e-05, |
|
"loss": 0.8114, |
|
"step": 52 |
|
}, |
|
{ |
|
"epoch": 53.0, |
|
"learning_rate": 2.6447368421052637e-05, |
|
"loss": 0.7796, |
|
"step": 53 |
|
}, |
|
{ |
|
"epoch": 54.0, |
|
"learning_rate": 2.6052631578947372e-05, |
|
"loss": 0.7848, |
|
"step": 54 |
|
}, |
|
{ |
|
"epoch": 55.0, |
|
"learning_rate": 2.5657894736842107e-05, |
|
"loss": 0.7618, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 56.0, |
|
"learning_rate": 2.526315789473684e-05, |
|
"loss": 0.7264, |
|
"step": 56 |
|
}, |
|
{ |
|
"epoch": 57.0, |
|
"learning_rate": 2.4868421052631583e-05, |
|
"loss": 0.7228, |
|
"step": 57 |
|
}, |
|
{ |
|
"epoch": 58.0, |
|
"learning_rate": 2.4473684210526318e-05, |
|
"loss": 0.7114, |
|
"step": 58 |
|
}, |
|
{ |
|
"epoch": 59.0, |
|
"learning_rate": 2.4078947368421056e-05, |
|
"loss": 0.7291, |
|
"step": 59 |
|
}, |
|
{ |
|
"epoch": 60.0, |
|
"learning_rate": 2.368421052631579e-05, |
|
"loss": 0.7543, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 60.0, |
|
"eval_accuracy_parking": NaN, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_accuracy_unparking": 0.8389015197753906, |
|
"eval_iou_parking": 0.0, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_iou_unparking": 0.8389015197753906, |
|
"eval_loss": 0.88163161277771, |
|
"eval_mean_accuracy": 0.8389015197753906, |
|
"eval_mean_iou": 0.2796338399251302, |
|
"eval_overall_accuracy": 0.8389015197753906, |
|
"eval_runtime": 0.4803, |
|
"eval_samples_per_second": 2.082, |
|
"eval_steps_per_second": 2.082, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 61.0, |
|
"learning_rate": 2.328947368421053e-05, |
|
"loss": 0.7015, |
|
"step": 61 |
|
}, |
|
{ |
|
"epoch": 62.0, |
|
"learning_rate": 2.2894736842105267e-05, |
|
"loss": 0.744, |
|
"step": 62 |
|
}, |
|
{ |
|
"epoch": 63.0, |
|
"learning_rate": 2.25e-05, |
|
"loss": 0.6976, |
|
"step": 63 |
|
}, |
|
{ |
|
"epoch": 64.0, |
|
"learning_rate": 2.2105263157894736e-05, |
|
"loss": 0.6847, |
|
"step": 64 |
|
}, |
|
{ |
|
"epoch": 65.0, |
|
"learning_rate": 2.1710526315789474e-05, |
|
"loss": 0.674, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 66.0, |
|
"learning_rate": 2.131578947368421e-05, |
|
"loss": 0.6956, |
|
"step": 66 |
|
}, |
|
{ |
|
"epoch": 67.0, |
|
"learning_rate": 2.0921052631578947e-05, |
|
"loss": 0.711, |
|
"step": 67 |
|
}, |
|
{ |
|
"epoch": 68.0, |
|
"learning_rate": 2.0526315789473685e-05, |
|
"loss": 0.7065, |
|
"step": 68 |
|
}, |
|
{ |
|
"epoch": 69.0, |
|
"learning_rate": 2.0131578947368423e-05, |
|
"loss": 0.6858, |
|
"step": 69 |
|
}, |
|
{ |
|
"epoch": 70.0, |
|
"learning_rate": 1.9736842105263158e-05, |
|
"loss": 0.6683, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 71.0, |
|
"learning_rate": 1.9342105263157896e-05, |
|
"loss": 0.7017, |
|
"step": 71 |
|
}, |
|
{ |
|
"epoch": 72.0, |
|
"learning_rate": 1.894736842105263e-05, |
|
"loss": 0.6584, |
|
"step": 72 |
|
}, |
|
{ |
|
"epoch": 73.0, |
|
"learning_rate": 1.855263157894737e-05, |
|
"loss": 0.7278, |
|
"step": 73 |
|
}, |
|
{ |
|
"epoch": 74.0, |
|
"learning_rate": 1.8157894736842104e-05, |
|
"loss": 0.678, |
|
"step": 74 |
|
}, |
|
{ |
|
"epoch": 75.0, |
|
"learning_rate": 1.7763157894736842e-05, |
|
"loss": 0.6417, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 76.0, |
|
"learning_rate": 1.736842105263158e-05, |
|
"loss": 0.6751, |
|
"step": 76 |
|
}, |
|
{ |
|
"epoch": 77.0, |
|
"learning_rate": 1.6973684210526318e-05, |
|
"loss": 0.6426, |
|
"step": 77 |
|
}, |
|
{ |
|
"epoch": 78.0, |
|
"learning_rate": 1.6578947368421053e-05, |
|
"loss": 0.7059, |
|
"step": 78 |
|
}, |
|
{ |
|
"epoch": 79.0, |
|
"learning_rate": 1.618421052631579e-05, |
|
"loss": 0.6402, |
|
"step": 79 |
|
}, |
|
{ |
|
"epoch": 80.0, |
|
"learning_rate": 1.5789473684210526e-05, |
|
"loss": 0.6537, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 80.0, |
|
"eval_accuracy_parking": NaN, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_accuracy_unparking": 0.8770866394042969, |
|
"eval_iou_parking": 0.0, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_iou_unparking": 0.8770866394042969, |
|
"eval_loss": 0.7174801826477051, |
|
"eval_mean_accuracy": 0.8770866394042969, |
|
"eval_mean_iou": 0.2923622131347656, |
|
"eval_overall_accuracy": 0.8770866394042969, |
|
"eval_runtime": 0.4859, |
|
"eval_samples_per_second": 2.058, |
|
"eval_steps_per_second": 2.058, |
|
"step": 80 |
|
} |
|
], |
|
"max_steps": 120, |
|
"num_train_epochs": 120, |
|
"total_flos": 2.24373352955904e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|