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": [
     "background",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:942bda4b6e51ae366e75f840087bc19c34a88a388aa7b36aa6d76b7576512067
-size 14891980
+oid sha256:c741f554f3910bd2815fedf11f9288bddae7297cd0f1688257b742b549fd8a13
+size 109465548

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:33d7f5d2f1397df6ffd7347ad14019b71828d48ff1f4ced54b496242c5b54e7d
+oid sha256:6ce97778098c2d53e2f42e36b63397dc42ad0063a5a0e0e4791a693539b84573
 size 4984