Training in progress, epoch 1

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "nvidia/mit-b4",
+  "_name_or_path": "nvidia/mit-b3",
   "architectures": [
     "SegformerForSemanticSegmentation"
   ],
@@ -8,8 +8,8 @@
   "decoder_hidden_size": 768,
   "depths": [
     3,
-    8,
-    27,
+    4,
+    18,
     3
   ],
   "downsampling_rates": [

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:05043ff0f5130061765177b42746f8373e555b499fe8104e0065155102a91cd0
-size 256313237
+oid sha256:3613adf27f3e8735cbee28b61f50acde2e0c6077bf4551a4cb4511d1b81d790d
+size 189131421

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:31fe7e8747dc4927facd82b3053c177ea558a71a2ddfa2284c24460076518749
-size 4027
+oid sha256:011a18d3e90c9151a97febdc63a918e80e0a1080dbb098b17e2866cd5ce14535
+size 3963