sam1120's picture
Training in progress, step 40
b6e3819
raw
history blame
7.41 kB
{
"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
}