Training in progress, step 20

config.json

@@ -1,16 +1,16 @@
 {
-  "_name_or_path": "nvidia/mit-b0",
+  "_name_or_path": "nvidia/mit-b4",
   "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,
+    8,
+    27,
+    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": "background",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f5f4ae340e123e2508fe126ad04120e0823902a00e33ba51d1f6eec03e79b543
-size 14931789
+oid sha256:6eb279cdf5280899451f7bc7703b65e8c95e27dd800657d1acc40d10ca851f7f
+size 256294101

runs/Jul07_21-27-45_9dd33c53a108/events.out.tfevents.1688765275.9dd33c53a108.976.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b38fc6ba91f8e62329065408991cbfa757050db2164a68034184c76d9e0446f0
+oid sha256:e2428dfac616fc08b5c48a42057ee6b72e0d6bbecf142d5b386c37b5c778427b
 size 3899