| { | |
| "best_metric": 0.5702608823776245, | |
| "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b0_7/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-06, | |
| "loss": 1.2178, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 1e-05, | |
| "loss": 1.2142, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "learning_rate": 1.5e-05, | |
| "loss": 1.2223, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "learning_rate": 2e-05, | |
| "loss": 1.208, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 1.67, | |
| "learning_rate": 2.5e-05, | |
| "loss": 1.2072, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "learning_rate": 3e-05, | |
| "loss": 1.2015, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 2.33, | |
| "learning_rate": 3.5000000000000004e-05, | |
| "loss": 1.1888, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 2.67, | |
| "learning_rate": 4e-05, | |
| "loss": 1.1768, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "learning_rate": 4.5e-05, | |
| "loss": 1.1543, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 3.33, | |
| "learning_rate": 5e-05, | |
| "loss": 1.1505, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 3.33, | |
| "eval_accuracy_dropoff": 0.9456199600248122, | |
| "eval_accuracy_undropoff": 0.2616499186280552, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.07032648592680048, | |
| "eval_iou_undropoff": 0.2615922132145178, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 1.1103304624557495, | |
| "eval_mean_accuracy": 0.6036349393264338, | |
| "eval_mean_iou": 0.11063956638043942, | |
| "eval_overall_accuracy": 0.2919347127278646, | |
| "eval_runtime": 1.7484, | |
| "eval_samples_per_second": 8.579, | |
| "eval_steps_per_second": 0.572, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 3.67, | |
| "learning_rate": 5.500000000000001e-05, | |
| "loss": 1.1405, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "learning_rate": 6e-05, | |
| "loss": 1.1039, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 4.33, | |
| "learning_rate": 6.500000000000001e-05, | |
| "loss": 1.1211, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 4.67, | |
| "learning_rate": 7.000000000000001e-05, | |
| "loss": 1.0914, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "learning_rate": 7.500000000000001e-05, | |
| "loss": 1.0583, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 5.33, | |
| "learning_rate": 8e-05, | |
| "loss": 1.0527, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 5.67, | |
| "learning_rate": 8.5e-05, | |
| "loss": 1.0227, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "learning_rate": 9e-05, | |
| "loss": 1.1078, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 6.33, | |
| "learning_rate": 8.973684210526316e-05, | |
| "loss": 1.0001, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 6.67, | |
| "learning_rate": 8.947368421052632e-05, | |
| "loss": 0.9635, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 6.67, | |
| "eval_accuracy_dropoff": 0.8177051025800078, | |
| "eval_accuracy_undropoff": 0.8763050644323176, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.2434656953279978, | |
| "eval_iou_undropoff": 0.8694137027963167, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 1.0114336013793945, | |
| "eval_mean_accuracy": 0.8470050835061627, | |
| "eval_mean_iou": 0.3709597993747715, | |
| "eval_overall_accuracy": 0.8737103780110677, | |
| "eval_runtime": 1.7139, | |
| "eval_samples_per_second": 8.752, | |
| "eval_steps_per_second": 0.583, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "learning_rate": 8.921052631578948e-05, | |
| "loss": 1.0405, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 7.33, | |
| "learning_rate": 8.894736842105263e-05, | |
| "loss": 0.9095, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 7.67, | |
| "learning_rate": 8.86842105263158e-05, | |
| "loss": 0.9104, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "learning_rate": 8.842105263157894e-05, | |
| "loss": 0.8736, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 8.33, | |
| "learning_rate": 8.815789473684211e-05, | |
| "loss": 0.8849, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 8.67, | |
| "learning_rate": 8.789473684210527e-05, | |
| "loss": 0.8143, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "learning_rate": 8.763157894736843e-05, | |
| "loss": 0.7846, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 9.33, | |
| "learning_rate": 8.736842105263158e-05, | |
| "loss": 0.7748, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 9.67, | |
| "learning_rate": 8.710526315789474e-05, | |
| "loss": 0.7513, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "learning_rate": 8.68421052631579e-05, | |
| "loss": 0.9358, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy_dropoff": 0.5848209157534404, | |
| "eval_accuracy_undropoff": 0.9605968730608304, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.3193744373731639, | |
| "eval_iou_undropoff": 0.9425019881457949, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 0.8242360353469849, | |
| "eval_mean_accuracy": 0.7727088944071354, | |
| "eval_mean_iou": 0.42062547517298626, | |
| "eval_overall_accuracy": 0.9439582824707031, | |
| "eval_runtime": 1.7189, | |
| "eval_samples_per_second": 8.726, | |
| "eval_steps_per_second": 0.582, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 10.33, | |
| "learning_rate": 8.657894736842107e-05, | |
| "loss": 0.7191, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 10.67, | |
| "learning_rate": 8.631578947368421e-05, | |
| "loss": 0.73, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 11.0, | |
| "learning_rate": 8.605263157894737e-05, | |
| "loss": 0.6168, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 11.33, | |
| "learning_rate": 8.578947368421054e-05, | |
| "loss": 0.6559, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 11.67, | |
| "learning_rate": 8.552631578947369e-05, | |
| "loss": 0.6277, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 12.0, | |
| "learning_rate": 8.526315789473684e-05, | |
| "loss": 0.6168, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 12.33, | |
| "learning_rate": 8.5e-05, | |
| "loss": 0.6095, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 12.67, | |
| "learning_rate": 8.473684210526316e-05, | |
| "loss": 0.5927, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 13.0, | |
| "learning_rate": 8.447368421052632e-05, | |
| "loss": 0.8162, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 13.33, | |
| "learning_rate": 8.421052631578948e-05, | |
| "loss": 0.579, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 13.33, | |
| "eval_accuracy_dropoff": 0.5401589817814231, | |
| "eval_accuracy_undropoff": 0.982923599779886, | |
| "eval_accuracy_unlabeled": NaN, | |
| "eval_iou_dropoff": 0.3951496002117638, | |
| "eval_iou_undropoff": 0.9624271055937778, | |
| "eval_iou_unlabeled": 0.0, | |
| "eval_loss": 0.5702608823776245, | |
| "eval_mean_accuracy": 0.7615412907806545, | |
| "eval_mean_iou": 0.4525255686018472, | |
| "eval_overall_accuracy": 0.9633188883463542, | |
| "eval_runtime": 1.8153, | |
| "eval_samples_per_second": 8.263, | |
| "eval_steps_per_second": 0.551, | |
| "step": 40 | |
| } | |
| ], | |
| "max_steps": 360, | |
| "num_train_epochs": 120, | |
| "total_flos": 8028359035453440.0, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |