|
{ |
|
"best_metric": 0.31975847482681274, |
|
"best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b0_7/checkpoint-80", |
|
"epoch": 26.666666666666668, |
|
"global_step": 80, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 5e-06, |
|
"loss": 1.2178, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1e-05, |
|
"loss": 1.2142, |
|
"step": 2 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 1.5e-05, |
|
"loss": 1.2223, |
|
"step": 3 |
|
}, |
|
{ |
|
"epoch": 1.33, |
|
"learning_rate": 2e-05, |
|
"loss": 1.208, |
|
"step": 4 |
|
}, |
|
{ |
|
"epoch": 1.67, |
|
"learning_rate": 2.5e-05, |
|
"loss": 1.2072, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 3e-05, |
|
"loss": 1.2015, |
|
"step": 6 |
|
}, |
|
{ |
|
"epoch": 2.33, |
|
"learning_rate": 3.5000000000000004e-05, |
|
"loss": 1.1888, |
|
"step": 7 |
|
}, |
|
{ |
|
"epoch": 2.67, |
|
"learning_rate": 4e-05, |
|
"loss": 1.1768, |
|
"step": 8 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"learning_rate": 4.5e-05, |
|
"loss": 1.1543, |
|
"step": 9 |
|
}, |
|
{ |
|
"epoch": 3.33, |
|
"learning_rate": 5e-05, |
|
"loss": 1.1505, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 3.33, |
|
"eval_accuracy_dropoff": 0.9456199600248122, |
|
"eval_accuracy_undropoff": 0.2616499186280552, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_iou_dropoff": 0.07032648592680048, |
|
"eval_iou_undropoff": 0.2615922132145178, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_loss": 1.1103304624557495, |
|
"eval_mean_accuracy": 0.6036349393264338, |
|
"eval_mean_iou": 0.11063956638043942, |
|
"eval_overall_accuracy": 0.2919347127278646, |
|
"eval_runtime": 1.7484, |
|
"eval_samples_per_second": 8.579, |
|
"eval_steps_per_second": 0.572, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 3.67, |
|
"learning_rate": 5.500000000000001e-05, |
|
"loss": 1.1405, |
|
"step": 11 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"learning_rate": 6e-05, |
|
"loss": 1.1039, |
|
"step": 12 |
|
}, |
|
{ |
|
"epoch": 4.33, |
|
"learning_rate": 6.500000000000001e-05, |
|
"loss": 1.1211, |
|
"step": 13 |
|
}, |
|
{ |
|
"epoch": 4.67, |
|
"learning_rate": 7.000000000000001e-05, |
|
"loss": 1.0914, |
|
"step": 14 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"learning_rate": 7.500000000000001e-05, |
|
"loss": 1.0583, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 5.33, |
|
"learning_rate": 8e-05, |
|
"loss": 1.0527, |
|
"step": 16 |
|
}, |
|
{ |
|
"epoch": 5.67, |
|
"learning_rate": 8.5e-05, |
|
"loss": 1.0227, |
|
"step": 17 |
|
}, |
|
{ |
|
"epoch": 6.0, |
|
"learning_rate": 9e-05, |
|
"loss": 1.1078, |
|
"step": 18 |
|
}, |
|
{ |
|
"epoch": 6.33, |
|
"learning_rate": 8.973684210526316e-05, |
|
"loss": 1.0001, |
|
"step": 19 |
|
}, |
|
{ |
|
"epoch": 6.67, |
|
"learning_rate": 8.947368421052632e-05, |
|
"loss": 0.9635, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 6.67, |
|
"eval_accuracy_dropoff": 0.8177051025800078, |
|
"eval_accuracy_undropoff": 0.8763050644323176, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_iou_dropoff": 0.2434656953279978, |
|
"eval_iou_undropoff": 0.8694137027963167, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_loss": 1.0114336013793945, |
|
"eval_mean_accuracy": 0.8470050835061627, |
|
"eval_mean_iou": 0.3709597993747715, |
|
"eval_overall_accuracy": 0.8737103780110677, |
|
"eval_runtime": 1.7139, |
|
"eval_samples_per_second": 8.752, |
|
"eval_steps_per_second": 0.583, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 7.0, |
|
"learning_rate": 8.921052631578948e-05, |
|
"loss": 1.0405, |
|
"step": 21 |
|
}, |
|
{ |
|
"epoch": 7.33, |
|
"learning_rate": 8.894736842105263e-05, |
|
"loss": 0.9095, |
|
"step": 22 |
|
}, |
|
{ |
|
"epoch": 7.67, |
|
"learning_rate": 8.86842105263158e-05, |
|
"loss": 0.9104, |
|
"step": 23 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"learning_rate": 8.842105263157894e-05, |
|
"loss": 0.8736, |
|
"step": 24 |
|
}, |
|
{ |
|
"epoch": 8.33, |
|
"learning_rate": 8.815789473684211e-05, |
|
"loss": 0.8849, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 8.67, |
|
"learning_rate": 8.789473684210527e-05, |
|
"loss": 0.8143, |
|
"step": 26 |
|
}, |
|
{ |
|
"epoch": 9.0, |
|
"learning_rate": 8.763157894736843e-05, |
|
"loss": 0.7846, |
|
"step": 27 |
|
}, |
|
{ |
|
"epoch": 9.33, |
|
"learning_rate": 8.736842105263158e-05, |
|
"loss": 0.7748, |
|
"step": 28 |
|
}, |
|
{ |
|
"epoch": 9.67, |
|
"learning_rate": 8.710526315789474e-05, |
|
"loss": 0.7513, |
|
"step": 29 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"learning_rate": 8.68421052631579e-05, |
|
"loss": 0.9358, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"eval_accuracy_dropoff": 0.5848209157534404, |
|
"eval_accuracy_undropoff": 0.9605968730608304, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_iou_dropoff": 0.3193744373731639, |
|
"eval_iou_undropoff": 0.9425019881457949, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_loss": 0.8242360353469849, |
|
"eval_mean_accuracy": 0.7727088944071354, |
|
"eval_mean_iou": 0.42062547517298626, |
|
"eval_overall_accuracy": 0.9439582824707031, |
|
"eval_runtime": 1.7189, |
|
"eval_samples_per_second": 8.726, |
|
"eval_steps_per_second": 0.582, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 10.33, |
|
"learning_rate": 8.657894736842107e-05, |
|
"loss": 0.7191, |
|
"step": 31 |
|
}, |
|
{ |
|
"epoch": 10.67, |
|
"learning_rate": 8.631578947368421e-05, |
|
"loss": 0.73, |
|
"step": 32 |
|
}, |
|
{ |
|
"epoch": 11.0, |
|
"learning_rate": 8.605263157894737e-05, |
|
"loss": 0.6168, |
|
"step": 33 |
|
}, |
|
{ |
|
"epoch": 11.33, |
|
"learning_rate": 8.578947368421054e-05, |
|
"loss": 0.6559, |
|
"step": 34 |
|
}, |
|
{ |
|
"epoch": 11.67, |
|
"learning_rate": 8.552631578947369e-05, |
|
"loss": 0.6277, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 12.0, |
|
"learning_rate": 8.526315789473684e-05, |
|
"loss": 0.6168, |
|
"step": 36 |
|
}, |
|
{ |
|
"epoch": 12.33, |
|
"learning_rate": 8.5e-05, |
|
"loss": 0.6095, |
|
"step": 37 |
|
}, |
|
{ |
|
"epoch": 12.67, |
|
"learning_rate": 8.473684210526316e-05, |
|
"loss": 0.5927, |
|
"step": 38 |
|
}, |
|
{ |
|
"epoch": 13.0, |
|
"learning_rate": 8.447368421052632e-05, |
|
"loss": 0.8162, |
|
"step": 39 |
|
}, |
|
{ |
|
"epoch": 13.33, |
|
"learning_rate": 8.421052631578948e-05, |
|
"loss": 0.579, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 13.33, |
|
"eval_accuracy_dropoff": 0.5401589817814231, |
|
"eval_accuracy_undropoff": 0.982923599779886, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_iou_dropoff": 0.3951496002117638, |
|
"eval_iou_undropoff": 0.9624271055937778, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_loss": 0.5702608823776245, |
|
"eval_mean_accuracy": 0.7615412907806545, |
|
"eval_mean_iou": 0.4525255686018472, |
|
"eval_overall_accuracy": 0.9633188883463542, |
|
"eval_runtime": 1.8153, |
|
"eval_samples_per_second": 8.263, |
|
"eval_steps_per_second": 0.551, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 13.67, |
|
"learning_rate": 8.394736842105263e-05, |
|
"loss": 0.5659, |
|
"step": 41 |
|
}, |
|
{ |
|
"epoch": 14.0, |
|
"learning_rate": 8.368421052631579e-05, |
|
"loss": 0.5447, |
|
"step": 42 |
|
}, |
|
{ |
|
"epoch": 14.33, |
|
"learning_rate": 8.342105263157896e-05, |
|
"loss": 0.5265, |
|
"step": 43 |
|
}, |
|
{ |
|
"epoch": 14.67, |
|
"learning_rate": 8.31578947368421e-05, |
|
"loss": 0.5263, |
|
"step": 44 |
|
}, |
|
{ |
|
"epoch": 15.0, |
|
"learning_rate": 8.289473684210527e-05, |
|
"loss": 0.4922, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 15.33, |
|
"learning_rate": 8.263157894736843e-05, |
|
"loss": 0.5099, |
|
"step": 46 |
|
}, |
|
{ |
|
"epoch": 15.67, |
|
"learning_rate": 8.236842105263158e-05, |
|
"loss": 0.4706, |
|
"step": 47 |
|
}, |
|
{ |
|
"epoch": 16.0, |
|
"learning_rate": 8.210526315789474e-05, |
|
"loss": 0.4809, |
|
"step": 48 |
|
}, |
|
{ |
|
"epoch": 16.33, |
|
"learning_rate": 8.18421052631579e-05, |
|
"loss": 0.4888, |
|
"step": 49 |
|
}, |
|
{ |
|
"epoch": 16.67, |
|
"learning_rate": 8.157894736842105e-05, |
|
"loss": 0.4411, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 16.67, |
|
"eval_accuracy_dropoff": 0.48720334505019874, |
|
"eval_accuracy_undropoff": 0.9888639114094216, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_iou_dropoff": 0.39281844198905264, |
|
"eval_iou_undropoff": 0.9659161382921264, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_loss": 0.4165554940700531, |
|
"eval_mean_accuracy": 0.7380336282298102, |
|
"eval_mean_iou": 0.452911526760393, |
|
"eval_overall_accuracy": 0.9666514078776042, |
|
"eval_runtime": 1.7457, |
|
"eval_samples_per_second": 8.593, |
|
"eval_steps_per_second": 0.573, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 17.0, |
|
"learning_rate": 8.131578947368422e-05, |
|
"loss": 0.4717, |
|
"step": 51 |
|
}, |
|
{ |
|
"epoch": 17.33, |
|
"learning_rate": 8.105263157894737e-05, |
|
"loss": 0.4523, |
|
"step": 52 |
|
}, |
|
{ |
|
"epoch": 17.67, |
|
"learning_rate": 8.078947368421052e-05, |
|
"loss": 0.4215, |
|
"step": 53 |
|
}, |
|
{ |
|
"epoch": 18.0, |
|
"learning_rate": 8.05263157894737e-05, |
|
"loss": 0.4007, |
|
"step": 54 |
|
}, |
|
{ |
|
"epoch": 18.33, |
|
"learning_rate": 8.026315789473685e-05, |
|
"loss": 0.4253, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 18.67, |
|
"learning_rate": 8e-05, |
|
"loss": 0.4153, |
|
"step": 56 |
|
}, |
|
{ |
|
"epoch": 19.0, |
|
"learning_rate": 7.973684210526316e-05, |
|
"loss": 0.6574, |
|
"step": 57 |
|
}, |
|
{ |
|
"epoch": 19.33, |
|
"learning_rate": 7.947368421052632e-05, |
|
"loss": 0.3899, |
|
"step": 58 |
|
}, |
|
{ |
|
"epoch": 19.67, |
|
"learning_rate": 7.921052631578948e-05, |
|
"loss": 0.4108, |
|
"step": 59 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"learning_rate": 7.894736842105263e-05, |
|
"loss": 0.4311, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 20.0, |
|
"eval_accuracy_dropoff": 0.44003721827831, |
|
"eval_accuracy_undropoff": 0.9911496700950386, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_iou_dropoff": 0.3694591258053316, |
|
"eval_iou_undropoff": 0.966086774739557, |
|
"eval_iou_unlabeled": NaN, |
|
"eval_loss": 0.38426053524017334, |
|
"eval_mean_accuracy": 0.7155934441866743, |
|
"eval_mean_iou": 0.6677729502724443, |
|
"eval_overall_accuracy": 0.966747538248698, |
|
"eval_runtime": 1.8089, |
|
"eval_samples_per_second": 8.292, |
|
"eval_steps_per_second": 0.553, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 20.33, |
|
"learning_rate": 7.868421052631579e-05, |
|
"loss": 0.3803, |
|
"step": 61 |
|
}, |
|
{ |
|
"epoch": 20.67, |
|
"learning_rate": 7.842105263157896e-05, |
|
"loss": 0.3828, |
|
"step": 62 |
|
}, |
|
{ |
|
"epoch": 21.0, |
|
"learning_rate": 7.815789473684212e-05, |
|
"loss": 0.5699, |
|
"step": 63 |
|
}, |
|
{ |
|
"epoch": 21.33, |
|
"learning_rate": 7.789473684210526e-05, |
|
"loss": 0.3803, |
|
"step": 64 |
|
}, |
|
{ |
|
"epoch": 21.67, |
|
"learning_rate": 7.763157894736843e-05, |
|
"loss": 0.3709, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 22.0, |
|
"learning_rate": 7.736842105263159e-05, |
|
"loss": 0.4738, |
|
"step": 66 |
|
}, |
|
{ |
|
"epoch": 22.33, |
|
"learning_rate": 7.710526315789474e-05, |
|
"loss": 0.3891, |
|
"step": 67 |
|
}, |
|
{ |
|
"epoch": 22.67, |
|
"learning_rate": 7.68421052631579e-05, |
|
"loss": 0.3576, |
|
"step": 68 |
|
}, |
|
{ |
|
"epoch": 23.0, |
|
"learning_rate": 7.657894736842105e-05, |
|
"loss": 0.5494, |
|
"step": 69 |
|
}, |
|
{ |
|
"epoch": 23.33, |
|
"learning_rate": 7.631578947368421e-05, |
|
"loss": 0.3437, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 23.33, |
|
"eval_accuracy_dropoff": 0.39948767431709054, |
|
"eval_accuracy_undropoff": 0.9917728653036201, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_iou_dropoff": 0.33924477871099273, |
|
"eval_iou_undropoff": 0.9649275553674269, |
|
"eval_iou_unlabeled": 0.0, |
|
"eval_loss": 0.35898175835609436, |
|
"eval_mean_accuracy": 0.6956302698103554, |
|
"eval_mean_iou": 0.43472411135947325, |
|
"eval_overall_accuracy": 0.9655476888020833, |
|
"eval_runtime": 1.8305, |
|
"eval_samples_per_second": 8.195, |
|
"eval_steps_per_second": 0.546, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 23.67, |
|
"learning_rate": 7.605263157894738e-05, |
|
"loss": 0.3447, |
|
"step": 71 |
|
}, |
|
{ |
|
"epoch": 24.0, |
|
"learning_rate": 7.578947368421052e-05, |
|
"loss": 0.3733, |
|
"step": 72 |
|
}, |
|
{ |
|
"epoch": 24.33, |
|
"learning_rate": 7.55263157894737e-05, |
|
"loss": 0.3481, |
|
"step": 73 |
|
}, |
|
{ |
|
"epoch": 24.67, |
|
"learning_rate": 7.526315789473685e-05, |
|
"loss": 0.3377, |
|
"step": 74 |
|
}, |
|
{ |
|
"epoch": 25.0, |
|
"learning_rate": 7.500000000000001e-05, |
|
"loss": 0.5311, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 25.33, |
|
"learning_rate": 7.473684210526316e-05, |
|
"loss": 0.3228, |
|
"step": 76 |
|
}, |
|
{ |
|
"epoch": 25.67, |
|
"learning_rate": 7.447368421052632e-05, |
|
"loss": 0.3248, |
|
"step": 77 |
|
}, |
|
{ |
|
"epoch": 26.0, |
|
"learning_rate": 7.421052631578948e-05, |
|
"loss": 0.5215, |
|
"step": 78 |
|
}, |
|
{ |
|
"epoch": 26.33, |
|
"learning_rate": 7.394736842105263e-05, |
|
"loss": 0.3395, |
|
"step": 79 |
|
}, |
|
{ |
|
"epoch": 26.67, |
|
"learning_rate": 7.368421052631579e-05, |
|
"loss": 0.3136, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 26.67, |
|
"eval_accuracy_dropoff": 0.33123693339766125, |
|
"eval_accuracy_undropoff": 0.9931389986088537, |
|
"eval_accuracy_unlabeled": NaN, |
|
"eval_iou_dropoff": 0.2885107958297481, |
|
"eval_iou_undropoff": 0.9632929658594979, |
|
"eval_iou_unlabeled": NaN, |
|
"eval_loss": 0.31975847482681274, |
|
"eval_mean_accuracy": 0.6621879660032575, |
|
"eval_mean_iou": 0.625901880844623, |
|
"eval_overall_accuracy": 0.9638313293457031, |
|
"eval_runtime": 1.7647, |
|
"eval_samples_per_second": 8.5, |
|
"eval_steps_per_second": 0.567, |
|
"step": 80 |
|
} |
|
], |
|
"max_steps": 360, |
|
"num_train_epochs": 120, |
|
"total_flos": 1.605671807090688e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|