Training in progress, epoch 1

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": "unlabeled",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0672864e903dc71ae8a07dfa48ecf162e0ed8b4c31fbafc5088e7b6a2e3d77fa
-size 14918708
+oid sha256:267f668db8c86dfb230698dc12abc3b938fcf89fb11b57d601726775db919a4e
+size 109545532

runs/Nov03_11-34-42_67ed50c70e15/events.out.tfevents.1730633691.67ed50c70e15.2774801.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:285574166589b358284f56de1d484521750031fbb1c5b057b58bda876ad2385b
+size 7471

runs/Nov03_11-38-18_67ed50c70e15/events.out.tfevents.1730633906.67ed50c70e15.2775701.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cdfd8ae618394c0ba7ef9ae078d8c1759d0ccce3bb5e92b854b8f3929cf5965
+size 12598

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f5743eb02fd0c2381ff872ba5bfa0f30b2579ca35002758764bd17afee50d3f2
+oid sha256:50046011f62cc7de2e7f9ef24b595e9e7d73d9d3ff863f3b48e4e09ce2e3d0d7
 size 5368