{ "best_metric": 1.038216471672058, "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b0_1/checkpoint-40", "epoch": 13.333333333333334, "global_step": 40, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.33, "learning_rate": 5e-07, "loss": 1.1127, "step": 1 }, { "epoch": 0.67, "learning_rate": 1e-06, "loss": 1.1148, "step": 2 }, { "epoch": 1.0, "learning_rate": 1.5e-06, "loss": 1.109, "step": 3 }, { "epoch": 1.33, "learning_rate": 2e-06, "loss": 1.1127, "step": 4 }, { "epoch": 1.67, "learning_rate": 2.5e-06, "loss": 1.1118, "step": 5 }, { "epoch": 2.0, "learning_rate": 3e-06, "loss": 1.1088, "step": 6 }, { "epoch": 2.33, "learning_rate": 3.5e-06, "loss": 1.1122, "step": 7 }, { "epoch": 2.67, "learning_rate": 4e-06, "loss": 1.1054, "step": 8 }, { "epoch": 3.0, "learning_rate": 4.5e-06, "loss": 1.1061, "step": 9 }, { "epoch": 3.33, "learning_rate": 5e-06, "loss": 1.1029, "step": 10 }, { "epoch": 3.33, "eval_accuracy_dropoff": 0.3332988719645278, "eval_accuracy_undropoff": 0.45771106945832574, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.04096214090185886, "eval_iou_undropoff": 0.4500892406005614, "eval_iou_unlabeled": 0.0, "eval_loss": 1.085178256034851, "eval_mean_accuracy": 0.39550497071142676, "eval_mean_iou": 0.16368379383414008, "eval_overall_accuracy": 0.4522023518880208, "eval_runtime": 1.7594, "eval_samples_per_second": 8.526, "eval_steps_per_second": 0.568, "step": 10 }, { "epoch": 3.67, "learning_rate": 5.500000000000001e-06, "loss": 1.1056, "step": 11 }, { "epoch": 4.0, "learning_rate": 6e-06, "loss": 1.1072, "step": 12 }, { "epoch": 4.33, "learning_rate": 6.5000000000000004e-06, "loss": 1.1039, "step": 13 }, { "epoch": 4.67, "learning_rate": 7e-06, "loss": 1.1023, "step": 14 }, { "epoch": 5.0, "learning_rate": 7.5e-06, "loss": 1.0945, "step": 15 }, { "epoch": 5.33, "learning_rate": 8e-06, "loss": 1.0956, "step": 16 }, { "epoch": 5.67, "learning_rate": 8.5e-06, "loss": 1.0946, "step": 17 }, { "epoch": 6.0, "learning_rate": 9e-06, "loss": 1.1014, "step": 18 }, { "epoch": 6.33, "learning_rate": 8.973684210526317e-06, "loss": 1.0905, "step": 19 }, { "epoch": 6.67, "learning_rate": 8.947368421052632e-06, "loss": 1.0856, "step": 20 }, { "epoch": 6.67, "eval_accuracy_dropoff": 0.5153065913111402, "eval_accuracy_undropoff": 0.5019142896372908, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.07605219279767873, "eval_iou_undropoff": 0.49721567971024616, "eval_iou_unlabeled": 0.0, "eval_loss": 1.076412320137024, "eval_mean_accuracy": 0.5086104404742156, "eval_mean_iou": 0.19108929083597495, "eval_overall_accuracy": 0.5025072733561198, "eval_runtime": 1.9083, "eval_samples_per_second": 7.861, "eval_steps_per_second": 0.524, "step": 20 }, { "epoch": 7.0, "learning_rate": 8.921052631578947e-06, "loss": 1.0941, "step": 21 }, { "epoch": 7.33, "learning_rate": 8.894736842105264e-06, "loss": 1.0807, "step": 22 }, { "epoch": 7.67, "learning_rate": 8.86842105263158e-06, "loss": 1.0792, "step": 23 }, { "epoch": 8.0, "learning_rate": 8.842105263157895e-06, "loss": 1.0701, "step": 24 }, { "epoch": 8.33, "learning_rate": 8.81578947368421e-06, "loss": 1.0735, "step": 25 }, { "epoch": 8.67, "learning_rate": 8.789473684210527e-06, "loss": 1.0715, "step": 26 }, { "epoch": 9.0, "learning_rate": 8.763157894736842e-06, "loss": 1.0614, "step": 27 }, { "epoch": 9.33, "learning_rate": 8.736842105263158e-06, "loss": 1.061, "step": 28 }, { "epoch": 9.67, "learning_rate": 8.710526315789473e-06, "loss": 1.0587, "step": 29 }, { "epoch": 10.0, "learning_rate": 8.68421052631579e-06, "loss": 1.0755, "step": 30 }, { "epoch": 10.0, "eval_accuracy_dropoff": 0.704510993176649, "eval_accuracy_undropoff": 0.5688494996876041, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.11037454693354545, "eval_iou_undropoff": 0.565151253990595, "eval_iou_unlabeled": 0.0, "eval_loss": 1.0611040592193604, "eval_mean_accuracy": 0.6366802464321266, "eval_mean_iou": 0.22517526697471346, "eval_overall_accuracy": 0.5748563130696615, "eval_runtime": 1.6608, "eval_samples_per_second": 9.032, "eval_steps_per_second": 0.602, "step": 30 }, { "epoch": 10.33, "learning_rate": 8.657894736842105e-06, "loss": 1.0575, "step": 31 }, { "epoch": 10.67, "learning_rate": 8.63157894736842e-06, "loss": 1.0559, "step": 32 }, { "epoch": 11.0, "learning_rate": 8.605263157894737e-06, "loss": 1.0358, "step": 33 }, { "epoch": 11.33, "learning_rate": 8.578947368421053e-06, "loss": 1.0525, "step": 34 }, { "epoch": 11.67, "learning_rate": 8.552631578947368e-06, "loss": 1.0447, "step": 35 }, { "epoch": 12.0, "learning_rate": 8.526315789473683e-06, "loss": 1.0542, "step": 36 }, { "epoch": 12.33, "learning_rate": 8.5e-06, "loss": 1.04, "step": 37 }, { "epoch": 12.67, "learning_rate": 8.473684210526317e-06, "loss": 1.0351, "step": 38 }, { "epoch": 13.0, "learning_rate": 8.447368421052632e-06, "loss": 1.0672, "step": 39 }, { "epoch": 13.33, "learning_rate": 8.421052631578947e-06, "loss": 1.0285, "step": 40 }, { "epoch": 13.33, "eval_accuracy_dropoff": 0.8494440232499368, "eval_accuracy_undropoff": 0.6478915140077891, "eval_accuracy_unlabeled": NaN, "eval_iou_dropoff": 0.14199210424243355, "eval_iou_undropoff": 0.6445086621047246, "eval_iou_unlabeled": 0.0, "eval_loss": 1.038216471672058, "eval_mean_accuracy": 0.748667768628863, "eval_mean_iou": 0.2621669221157194, "eval_overall_accuracy": 0.6568158467610677, "eval_runtime": 1.6441, "eval_samples_per_second": 9.123, "eval_steps_per_second": 0.608, "step": 40 } ], "max_steps": 360, "num_train_epochs": 120, "total_flos": 8028359035453440.0, "trial_name": null, "trial_params": null }