{ "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 }