Training in progress, step 20

config.json

@@ -1,31 +1,25 @@
 {
-  "_name_or_path": "nvidia/mit-b0",
+  "_name_or_path": "Matthijs/deeplabv3-mobilevit-small",
   "architectures": [
-    "SegformerForSemanticSegmentation"
+    "MobileViTForSemanticSegmentation"
+  ],
+  "aspp_dropout_prob": 0.1,
+  "aspp_out_channels": 256,
+  "atrous_rates": [
+    6,
+    12,
+    18
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
-  "decoder_hidden_size": 256,
-  "depths": [
-    2,
-    2,
-    2,
-    2
-  ],
-  "downsampling_rates": [
-    1,
-    4,
-    8,
-    16
-  ],
-  "drop_path_rate": 0.1,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.0,
+  "conv_kernel_size": 3,
+  "expand_ratio": 4.0,
+  "hidden_act": "silu",
+  "hidden_dropout_prob": 0.1,
   "hidden_sizes": [
-    32,
-    64,
-    160,
-    256
+    144,
+    192,
+    240
   ],
   "id2label": {
     "0": "void",
@@ -48,7 +42,7 @@
     "17": "Sky",
     "18": "Grass"
   },
-  "image_size": 224,
+  "image_size": 512,
   "initializer_range": 0.02,
   "label2id": {
     "Asphalt": 1,
@@ -71,42 +65,24 @@
     "Water": 4,
     "void": 0
   },
-  "layer_norm_eps": 1e-06,
-  "mlp_ratios": [
-    4,
-    4,
-    4,
-    4
-  ],
-  "model_type": "segformer",
-  "num_attention_heads": [
-    1,
-    2,
-    5,
-    8
+  "layer_norm_eps": 1e-05,
+  "mlp_ratio": 2.0,
+  "model_type": "mobilevit",
+  "neck_hidden_sizes": [
+    16,
+    32,
+    64,
+    96,
+    128,
+    160,
+    640
   ],
+  "num_attention_heads": 4,
   "num_channels": 3,
-  "num_encoder_blocks": 4,
-  "patch_sizes": [
-    7,
-    3,
-    3,
-    3
-  ],
-  "reshape_last_stage": true,
+  "output_stride": 16,
+  "patch_size": 2,
+  "qkv_bias": true,
   "semantic_loss_ignore_index": 255,
-  "sr_ratios": [
-    8,
-    4,
-    2,
-    1
-  ],
-  "strides": [
-    4,
-    2,
-    2,
-    2
-  ],
   "torch_dtype": "float32",
   "transformers_version": "4.42.3"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f53c380006dea77d7650b2d9d3b2027aed817df45df50fc0904914029eef5d48
-size 14902260
+oid sha256:549b7d69f035a514d39dca028a3f8724cf0a687494bfc7397097ca84a51b9f6c
+size 25536516

runs/Jul17_21-09-27_Jian-PC/events.out.tfevents.1721214568.Jian-PC.10912.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:53492c9b72f91e5e8be5f3ddae16f57ed2206affbebd01cb04067d926f4cef93
+oid sha256:075afd06ab61c36048a7c1af4d9c4a3ed272b4e1273f469f453ad7379ec85dec
 size 5176