| { | |
| "best_metric": 0.47930747270584106, | |
| "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGBD-b0_6/checkpoint-40", | |
| "epoch": 20.0, | |
| "global_step": 40, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.5, | |
| "learning_rate": 5.833333333333332e-06, | |
| "loss": 1.079, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "learning_rate": 1.1666666666666665e-05, | |
| "loss": 1.0795, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 1.5, | |
| "learning_rate": 1.75e-05, | |
| "loss": 1.0738, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "learning_rate": 2.333333333333333e-05, | |
| "loss": 1.0662, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 2.5, | |
| "learning_rate": 2.9166666666666666e-05, | |
| "loss": 1.0593, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "learning_rate": 3.5e-05, | |
| "loss": 1.0514, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 3.5, | |
| "learning_rate": 4.0833333333333334e-05, | |
| "loss": 1.0365, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "learning_rate": 4.666666666666666e-05, | |
| "loss": 1.0287, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 4.5, | |
| "learning_rate": 5.2499999999999995e-05, | |
| "loss": 1.0123, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "learning_rate": 5.833333333333333e-05, | |
| "loss": 0.9975, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy_dropoff": 0.6267483199155944, | |
| "eval_accuracy_undropoff": 0.7225725563868626, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.128993031460863, | |
| "eval_iou_undropoff": 0.7166682161261027, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 1.047046422958374, | |
| "eval_mean_accuracy": 0.6746604381512284, | |
| "eval_mean_iou": 0.28188708252898853, | |
| "eval_overall_accuracy": 0.7185882568359375, | |
| "eval_runtime": 2.9658, | |
| "eval_samples_per_second": 6.744, | |
| "eval_steps_per_second": 0.674, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 5.5, | |
| "learning_rate": 6.416666666666666e-05, | |
| "loss": 0.9837, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "learning_rate": 7e-05, | |
| "loss": 0.9765, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 6.5, | |
| "learning_rate": 6.969298245614035e-05, | |
| "loss": 0.9655, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "learning_rate": 6.938596491228069e-05, | |
| "loss": 0.9246, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 7.5, | |
| "learning_rate": 6.907894736842105e-05, | |
| "loss": 0.8986, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "learning_rate": 6.877192982456139e-05, | |
| "loss": 0.8794, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 8.5, | |
| "learning_rate": 6.846491228070175e-05, | |
| "loss": 0.8586, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "learning_rate": 6.81578947368421e-05, | |
| "loss": 0.8485, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 9.5, | |
| "learning_rate": 6.785087719298245e-05, | |
| "loss": 0.8307, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "learning_rate": 6.75438596491228e-05, | |
| "loss": 0.8329, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy_dropoff": 0.01171586504277621, | |
| "eval_accuracy_undropoff": 0.9934380587814448, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.010567823996491185, | |
| "eval_iou_undropoff": 0.9526087857573317, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 0.8435441255569458, | |
| "eval_mean_accuracy": 0.5025769619121104, | |
| "eval_mean_iou": 0.3210588699179409, | |
| "eval_overall_accuracy": 0.9526187896728515, | |
| "eval_runtime": 2.8522, | |
| "eval_samples_per_second": 7.012, | |
| "eval_steps_per_second": 0.701, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 10.5, | |
| "learning_rate": 6.723684210526316e-05, | |
| "loss": 0.8067, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 11.0, | |
| "learning_rate": 6.69298245614035e-05, | |
| "loss": 0.7975, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 11.5, | |
| "learning_rate": 6.662280701754386e-05, | |
| "loss": 0.7694, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 12.0, | |
| "learning_rate": 6.63157894736842e-05, | |
| "loss": 0.7727, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 12.5, | |
| "learning_rate": 6.600877192982455e-05, | |
| "loss": 0.757, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 13.0, | |
| "learning_rate": 6.570175438596491e-05, | |
| "loss": 0.7564, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 13.5, | |
| "learning_rate": 6.539473684210525e-05, | |
| "loss": 0.7134, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 14.0, | |
| "learning_rate": 6.508771929824561e-05, | |
| "loss": 0.7425, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 14.5, | |
| "learning_rate": 6.478070175438596e-05, | |
| "loss": 0.7061, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 15.0, | |
| "learning_rate": 6.447368421052631e-05, | |
| "loss": 0.6857, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 15.0, | |
| "eval_accuracy_dropoff": 0.0006468038257758205, | |
| "eval_accuracy_undropoff": 0.998134683679328, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.0006266750223336311, | |
| "eval_iou_undropoff": 0.9566587236175594, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 0.6183835864067078, | |
| "eval_mean_accuracy": 0.49939074375255194, | |
| "eval_mean_iou": 0.3190951328799643, | |
| "eval_overall_accuracy": 0.9566598892211914, | |
| "eval_runtime": 3.0847, | |
| "eval_samples_per_second": 6.484, | |
| "eval_steps_per_second": 0.648, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 15.5, | |
| "learning_rate": 6.416666666666666e-05, | |
| "loss": 0.6817, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 16.0, | |
| "learning_rate": 6.3859649122807e-05, | |
| "loss": 0.7188, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 16.5, | |
| "learning_rate": 6.355263157894736e-05, | |
| "loss": 0.6683, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 17.0, | |
| "learning_rate": 6.324561403508772e-05, | |
| "loss": 0.6371, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 17.5, | |
| "learning_rate": 6.293859649122806e-05, | |
| "loss": 0.651, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 18.0, | |
| "learning_rate": 6.263157894736842e-05, | |
| "loss": 0.6381, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 18.5, | |
| "learning_rate": 6.232456140350877e-05, | |
| "loss": 0.6075, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 19.0, | |
| "learning_rate": 6.201754385964911e-05, | |
| "loss": 0.6391, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 19.5, | |
| "learning_rate": 6.171052631578947e-05, | |
| "loss": 0.6282, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "learning_rate": 6.140350877192981e-05, | |
| "loss": 0.5913, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "eval_accuracy_dropoff": 0.0005458840799100897, | |
| "eval_accuracy_undropoff": 0.9988294259470615, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.0005386297079178567, | |
| "eval_iou_undropoff": 0.957320579747961, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 0.47930747270584106, | |
| "eval_mean_accuracy": 0.4996876550134858, | |
| "eval_mean_iou": 0.3192864031519596, | |
| "eval_overall_accuracy": 0.9573215484619141, | |
| "eval_runtime": 2.984, | |
| "eval_samples_per_second": 6.702, | |
| "eval_steps_per_second": 0.67, | |
| "step": 40 | |
| } | |
| ], | |
| "max_steps": 240, | |
| "num_train_epochs": 120, | |
| "total_flos": 2.035958276947968e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |