Model save

checkpoint-40/config.json

@@ -1,80 +0,0 @@
-{
-  "_name_or_path": "nvidia/mit-b0",
-  "architectures": [
-    "SegformerForSemanticSegmentation"
-  ],
-  "attention_probs_dropout_prob": 0.0,
-  "classifier_dropout_prob": 0.1,
-  "decoder_hidden_size": 256,
-  "depths": [
-    2,
-    2,
-    2,
-    2
-  ],
-  "downsampling_rates": [
-    1,
-    4,
-    8,
-    16
-  ],
-  "drop_path_rate": 0.1,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.0,
-  "hidden_sizes": [
-    32,
-    64,
-    160,
-    256
-  ],
-  "id2label": {
-    "0": "unlabeled",
-    "1": "safe",
-    "2": "unsafe"
-  },
-  "image_size": 224,
-  "initializer_range": 0.02,
-  "label2id": {
-    "safe": 1,
-    "unlabeled": 0,
-    "unsafe": 2
-  },
-  "layer_norm_eps": 1e-06,
-  "mlp_ratios": [
-    4,
-    4,
-    4,
-    4
-  ],
-  "model_type": "segformer",
-  "num_attention_heads": [
-    1,
-    2,
-    5,
-    8
-  ],
-  "num_channels": 6,
-  "num_encoder_blocks": 4,
-  "patch_sizes": [
-    7,
-    3,
-    3,
-    3
-  ],
-  "reshape_last_stage": true,
-  "semantic_loss_ignore_index": 255,
-  "sr_ratios": [
-    8,
-    4,
-    2,
-    1
-  ],
-  "strides": [
-    4,
-    2,
-    2,
-    2
-  ],
-  "torch_dtype": "float32",
-  "transformers_version": "4.30.2"
-}

checkpoint-40/optimizer.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8eaf7ee4497829e11ba490ac0708806f4501908a3f299a997814b0dd4e4eeff8
-size 29876485

checkpoint-40/pytorch_model.bin

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5a1bb3851f52bd0b25782c9266067daa6232480a4fa6c1a39ab3a6da3e6cc468
-size 14951629

checkpoint-40/rng_state.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8069c3d91db9479f735c271dc368d96cf4daf90bfc98cd01aaab53534e9db295
-size 14575

checkpoint-40/scheduler.pt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:fb456e88934ce76c40f1da9d68d3c9076281aae82aa3b41e238953dab6fce1de
-size 627

checkpoint-40/trainer_state.json

@@ -1,324 +0,0 @@
-{
-  "best_metric": 0.6213392615318298,
-  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-train-SF30-RGBD-b0/checkpoint-40",
-  "epoch": 20.0,
-  "global_step": 40,
-  "is_hyper_param_search": false,
-  "is_local_process_zero": true,
-  "is_world_process_zero": true,
-  "log_history": [
-    {
-      "epoch": 0.5,
-      "learning_rate": 5e-06,
-      "loss": 1.0565,
-      "step": 1
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1e-05,
-      "loss": 1.0485,
-      "step": 2
-    },
-    {
-      "epoch": 1.5,
-      "learning_rate": 1.5e-05,
-      "loss": 1.0559,
-      "step": 3
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 2e-05,
-      "loss": 1.0501,
-      "step": 4
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 2.5e-05,
-      "loss": 1.0391,
-      "step": 5
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 3e-05,
-      "loss": 1.0435,
-      "step": 6
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 3.5e-05,
-      "loss": 1.0235,
-      "step": 7
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 4e-05,
-      "loss": 1.0155,
-      "step": 8
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 4.5e-05,
-      "loss": 1.0107,
-      "step": 9
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 5e-05,
-      "loss": 0.9925,
-      "step": 10
-    },
-    {
-      "epoch": 5.0,
-      "eval_accuracy_safe": 0.16253187340632969,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9084522788185269,
-      "eval_iou_safe": 0.04617705465733694,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8841402740524446,
-      "eval_loss": 1.0611684322357178,
-      "eval_mean_accuracy": 0.5354920761124282,
-      "eval_mean_iou": 0.3101057762365938,
-      "eval_overall_accuracy": 0.8847389221191406,
-      "eval_runtime": 2.0635,
-      "eval_samples_per_second": 3.877,
-      "eval_steps_per_second": 0.485,
-      "step": 10
-    },
-    {
-      "epoch": 5.5,
-      "learning_rate": 4.973684210526316e-05,
-      "loss": 0.9711,
-      "step": 11
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 4.9473684210526315e-05,
-      "loss": 0.9794,
-      "step": 12
-    },
-    {
-      "epoch": 6.5,
-      "learning_rate": 4.921052631578947e-05,
-      "loss": 0.9568,
-      "step": 13
-    },
-    {
-      "epoch": 7.0,
-      "learning_rate": 4.8947368421052635e-05,
-      "loss": 0.9109,
-      "step": 14
-    },
-    {
-      "epoch": 7.5,
-      "learning_rate": 4.868421052631579e-05,
-      "loss": 0.9157,
-      "step": 15
-    },
-    {
-      "epoch": 8.0,
-      "learning_rate": 4.842105263157895e-05,
-      "loss": 0.8739,
-      "step": 16
-    },
-    {
-      "epoch": 8.5,
-      "learning_rate": 4.8157894736842105e-05,
-      "loss": 0.8838,
-      "step": 17
-    },
-    {
-      "epoch": 9.0,
-      "learning_rate": 4.789473684210526e-05,
-      "loss": 0.9098,
-      "step": 18
-    },
-    {
-      "epoch": 9.5,
-      "learning_rate": 4.7631578947368424e-05,
-      "loss": 0.8599,
-      "step": 19
-    },
-    {
-      "epoch": 10.0,
-      "learning_rate": 4.736842105263158e-05,
-      "loss": 0.8589,
-      "step": 20
-    },
-    {
-      "epoch": 10.0,
-      "eval_accuracy_safe": 0.05294735263236838,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9833241565303213,
-      "eval_iou_safe": 0.03730081575721712,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9536688093648232,
-      "eval_loss": 0.9441336989402771,
-      "eval_mean_accuracy": 0.5181357545813449,
-      "eval_mean_iou": 0.33032320837401347,
-      "eval_overall_accuracy": 0.9537467956542969,
-      "eval_runtime": 1.9246,
-      "eval_samples_per_second": 4.157,
-      "eval_steps_per_second": 0.52,
-      "step": 20
-    },
-    {
-      "epoch": 10.5,
-      "learning_rate": 4.7105263157894744e-05,
-      "loss": 0.8438,
-      "step": 21
-    },
-    {
-      "epoch": 11.0,
-      "learning_rate": 4.68421052631579e-05,
-      "loss": 0.828,
-      "step": 22
-    },
-    {
-      "epoch": 11.5,
-      "learning_rate": 4.657894736842106e-05,
-      "loss": 0.8353,
-      "step": 23
-    },
-    {
-      "epoch": 12.0,
-      "learning_rate": 4.6315789473684214e-05,
-      "loss": 0.8821,
-      "step": 24
-    },
-    {
-      "epoch": 12.5,
-      "learning_rate": 4.605263157894737e-05,
-      "loss": 0.8043,
-      "step": 25
-    },
-    {
-      "epoch": 13.0,
-      "learning_rate": 4.5789473684210527e-05,
-      "loss": 0.7807,
-      "step": 26
-    },
-    {
-      "epoch": 13.5,
-      "learning_rate": 4.552631578947369e-05,
-      "loss": 0.7855,
-      "step": 27
-    },
-    {
-      "epoch": 14.0,
-      "learning_rate": 4.5263157894736846e-05,
-      "loss": 0.7243,
-      "step": 28
-    },
-    {
-      "epoch": 14.5,
-      "learning_rate": 4.5e-05,
-      "loss": 0.7318,
-      "step": 29
-    },
-    {
-      "epoch": 15.0,
-      "learning_rate": 4.473684210526316e-05,
-      "loss": 0.7016,
-      "step": 30
-    },
-    {
-      "epoch": 15.0,
-      "eval_accuracy_safe": 0.017234138293085347,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9965333354346406,
-      "eval_iou_safe": 0.016899046946699612,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.965381728938365,
-      "eval_loss": 0.7763598561286926,
-      "eval_mean_accuracy": 0.506883736863863,
-      "eval_mean_iou": 0.32742692529502154,
-      "eval_overall_accuracy": 0.9654006958007812,
-      "eval_runtime": 1.8724,
-      "eval_samples_per_second": 4.273,
-      "eval_steps_per_second": 0.534,
-      "step": 30
-    },
-    {
-      "epoch": 15.5,
-      "learning_rate": 4.4473684210526316e-05,
-      "loss": 0.6872,
-      "step": 31
-    },
-    {
-      "epoch": 16.0,
-      "learning_rate": 4.421052631578947e-05,
-      "loss": 0.6695,
-      "step": 32
-    },
-    {
-      "epoch": 16.5,
-      "learning_rate": 4.394736842105263e-05,
-      "loss": 0.6728,
-      "step": 33
-    },
-    {
-      "epoch": 17.0,
-      "learning_rate": 4.368421052631579e-05,
-      "loss": 0.757,
-      "step": 34
-    },
-    {
-      "epoch": 17.5,
-      "learning_rate": 4.342105263157895e-05,
-      "loss": 0.6676,
-      "step": 35
-    },
-    {
-      "epoch": 18.0,
-      "learning_rate": 4.3157894736842105e-05,
-      "loss": 0.6491,
-      "step": 36
-    },
-    {
-      "epoch": 18.5,
-      "learning_rate": 4.289473684210527e-05,
-      "loss": 0.6541,
-      "step": 37
-    },
-    {
-      "epoch": 19.0,
-      "learning_rate": 4.2631578947368425e-05,
-      "loss": 0.6635,
-      "step": 38
-    },
-    {
-      "epoch": 19.5,
-      "learning_rate": 4.236842105263158e-05,
-      "loss": 0.6657,
-      "step": 39
-    },
-    {
-      "epoch": 20.0,
-      "learning_rate": 4.210526315789474e-05,
-      "loss": 0.6093,
-      "step": 40
-    },
-    {
-      "epoch": 20.0,
-      "eval_accuracy_safe": 0.053772311384430775,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9900870827714798,
-      "eval_iou_safe": 0.04153054841176062,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9602530036058077,
-      "eval_loss": 0.6213392615318298,
-      "eval_mean_accuracy": 0.5219296970779553,
-      "eval_mean_iou": 0.33392785067252273,
-      "eval_overall_accuracy": 0.9603209495544434,
-      "eval_runtime": 1.9138,
-      "eval_samples_per_second": 4.18,
-      "eval_steps_per_second": 0.523,
-      "step": 40
-    }
-  ],
-  "max_steps": 200,
-  "num_train_epochs": 100,
-  "total_flos": 1.40410915651584e+16,
-  "trial_name": null,
-  "trial_params": null
-}

checkpoint-40/training_args.bin

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c4054f413ab6333d67080b8665aa1867742d0278adc7d6669a9b148ead4b21c7
-size 4155

runs/Feb09_07-20-24_robovision.csres.utexas.edu/events.out.tfevents.1707484830.robovision.csres.utexas.edu.1829578.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d2030a966ccc33214679797d5ffb1ba32ed496b99c353bd6890db4b96eaa5dda
-size 51233
+oid sha256:3ae255eea78f7c2ee2e24d79b5414b2222c6cecdb51f5123baf7b67a0db6f919
+size 51587