Training in progress, step 20

config.json

@@ -1,16 +1,16 @@
 {
-  "_name_or_path": "nvidia/mit-b0",
+  "_name_or_path": "nvidia/mit-b2",
   "architectures": [
     "SegformerForSemanticSegmentation"
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
-  "decoder_hidden_size": 256,
+  "decoder_hidden_size": 768,
   "depths": [
-    2,
-    2,
-    2,
-    2
+    3,
+    4,
+    6,
+    3
   ],
   "downsampling_rates": [
     1,
@@ -22,10 +22,10 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_sizes": [
-    32,
     64,
-    160,
-    256
+    128,
+    320,
+    512
   ],
   "id2label": {
     "0": "runway"

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e3c2b24361208e76a9d53c27c9c7e7a040429417988a6764fcb85e60b96e192a
-size 14883748
+oid sha256:1539354171637801fb8d435034ffa974ff8cbe996a227c6edacedeedcb8762e0
+size 109440940

runs/Nov13_23-57-04_7a9e0d4dfe3e/events.out.tfevents.1731542225.7a9e0d4dfe3e.46843.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f4b42551cc85730c941de3b1838069d009566d9fb656b4485e11d41829695bea
+oid sha256:8fb652429ea03c1d98daf295e1b2694897d0e149ecc91fcf5ef2e9640f1a9c50
 size 5304