| { | |
| "best_metric": 0.6001297235488892, | |
| "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b0_6/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": 3.888888888888889e-06, | |
| "loss": 1.1755, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 7.777777777777777e-06, | |
| "loss": 1.1749, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "learning_rate": 1.1666666666666665e-05, | |
| "loss": 1.183, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "learning_rate": 1.5555555555555555e-05, | |
| "loss": 1.1718, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 1.67, | |
| "learning_rate": 1.9444444444444442e-05, | |
| "loss": 1.1706, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "learning_rate": 2.333333333333333e-05, | |
| "loss": 1.1696, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 2.33, | |
| "learning_rate": 2.722222222222222e-05, | |
| "loss": 1.1558, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 2.67, | |
| "learning_rate": 3.111111111111111e-05, | |
| "loss": 1.1489, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "learning_rate": 3.5e-05, | |
| "loss": 1.135, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 3.33, | |
| "learning_rate": 3.8888888888888884e-05, | |
| "loss": 1.1234, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 3.33, | |
| "eval_accuracy_dropoff": 0.772015071105291, | |
| "eval_accuracy_undropoff": 0.353825865102452, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.1801195043477445, | |
| "eval_iou_undropoff": 0.3536296682807036, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 1.097259283065796, | |
| "eval_mean_accuracy": 0.5629204681038715, | |
| "eval_mean_iou": 0.17791639087614938, | |
| "eval_overall_accuracy": 0.37234242757161456, | |
| "eval_runtime": 1.873, | |
| "eval_samples_per_second": 8.008, | |
| "eval_steps_per_second": 0.534, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 3.67, | |
| "learning_rate": 4.277777777777778e-05, | |
| "loss": 1.1168, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "learning_rate": 4.666666666666666e-05, | |
| "loss": 1.0895, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 4.33, | |
| "learning_rate": 5.055555555555555e-05, | |
| "loss": 1.1027, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 4.67, | |
| "learning_rate": 5.444444444444444e-05, | |
| "loss": 1.0795, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "learning_rate": 5.833333333333333e-05, | |
| "loss": 1.0598, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 5.33, | |
| "learning_rate": 6.222222222222222e-05, | |
| "loss": 1.0449, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 5.67, | |
| "learning_rate": 6.611111111111111e-05, | |
| "loss": 1.0268, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "learning_rate": 7e-05, | |
| "loss": 1.0843, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 6.33, | |
| "learning_rate": 6.97953216374269e-05, | |
| "loss": 1.0029, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 6.67, | |
| "learning_rate": 6.95906432748538e-05, | |
| "loss": 0.975, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 6.67, | |
| "eval_accuracy_dropoff": 0.8258609598639924, | |
| "eval_accuracy_undropoff": 0.8102080545984994, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.24279987572018127, | |
| "eval_iou_undropoff": 0.8068537963656117, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 1.0260422229766846, | |
| "eval_mean_accuracy": 0.8180345072312459, | |
| "eval_mean_iou": 0.34988455736193097, | |
| "eval_overall_accuracy": 0.810901133219401, | |
| "eval_runtime": 1.6566, | |
| "eval_samples_per_second": 9.055, | |
| "eval_steps_per_second": 0.604, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "learning_rate": 6.938596491228069e-05, | |
| "loss": 1.0358, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 7.33, | |
| "learning_rate": 6.91812865497076e-05, | |
| "loss": 0.9361, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 7.67, | |
| "learning_rate": 6.89766081871345e-05, | |
| "loss": 0.9311, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "learning_rate": 6.877192982456139e-05, | |
| "loss": 0.9149, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 8.33, | |
| "learning_rate": 6.85672514619883e-05, | |
| "loss": 0.9056, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 8.67, | |
| "learning_rate": 6.83625730994152e-05, | |
| "loss": 0.8637, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "learning_rate": 6.81578947368421e-05, | |
| "loss": 0.7991, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 9.33, | |
| "learning_rate": 6.7953216374269e-05, | |
| "loss": 0.8143, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 9.67, | |
| "learning_rate": 6.774853801169589e-05, | |
| "loss": 0.7999, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "learning_rate": 6.75438596491228e-05, | |
| "loss": 0.9464, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy_dropoff": 0.5212052289383601, | |
| "eval_accuracy_undropoff": 0.9700488444545206, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.3384996437670423, | |
| "eval_iou_undropoff": 0.9507046257182781, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 0.8130465149879456, | |
| "eval_mean_accuracy": 0.7456270366964404, | |
| "eval_mean_iou": 0.4297347564951068, | |
| "eval_overall_accuracy": 0.9501749674479166, | |
| "eval_runtime": 1.7097, | |
| "eval_samples_per_second": 8.773, | |
| "eval_steps_per_second": 0.585, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 10.33, | |
| "learning_rate": 6.733918128654971e-05, | |
| "loss": 0.763, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 10.67, | |
| "learning_rate": 6.71345029239766e-05, | |
| "loss": 0.7591, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 11.0, | |
| "learning_rate": 6.69298245614035e-05, | |
| "loss": 0.6623, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 11.33, | |
| "learning_rate": 6.672514619883041e-05, | |
| "loss": 0.7138, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 11.67, | |
| "learning_rate": 6.65204678362573e-05, | |
| "loss": 0.6695, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 12.0, | |
| "learning_rate": 6.63157894736842e-05, | |
| "loss": 0.6987, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 12.33, | |
| "learning_rate": 6.611111111111111e-05, | |
| "loss": 0.6494, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 12.67, | |
| "learning_rate": 6.5906432748538e-05, | |
| "loss": 0.6317, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 13.0, | |
| "learning_rate": 6.570175438596491e-05, | |
| "loss": 0.834, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 13.33, | |
| "learning_rate": 6.54970760233918e-05, | |
| "loss": 0.6167, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 13.33, | |
| "eval_accuracy_dropoff": 0.5047671560181037, | |
| "eval_accuracy_undropoff": 0.9829004494881923, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.3743041743152479, | |
| "eval_iou_undropoff": 0.961015278770039, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 0.6001297235488892, | |
| "eval_mean_accuracy": 0.743833802753148, | |
| "eval_mean_iou": 0.44510648436176226, | |
| "eval_overall_accuracy": 0.9617296854654948, | |
| "eval_runtime": 1.7423, | |
| "eval_samples_per_second": 8.609, | |
| "eval_steps_per_second": 0.574, | |
| "step": 40 | |
| } | |
| ], | |
| "max_steps": 360, | |
| "num_train_epochs": 120, | |
| "total_flos": 8028359035453440.0, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |