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:05ce2168ead8dc39259c859ad739de70fee840003a6b783e4bb9c515b50d6d23
-size 256307093
+oid sha256:8fc2bc54a10d57627d6e18abc0867252d16c97f65f72da71c1ffbbe44b9de384
+size 189125277

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:85fa4b8dae839d3740eb3de3a8b886a848fcdcad2ed7f7b62c9c83e188a31f8a
-size 4027
+oid sha256:74d3793cc13830249269bd08593c624bb89e240a76a130fb98d4a5617982b7a2
+size 3963