Training in progress steps 4

config.json

@@ -1,16 +1,16 @@
 {
-  "_name_or_path": "/home/administrator/Omar280X/news/memes_line_seg_b5_newspaper_generator_x10_30epochs/",
+  "_name_or_path": "nvidia/segformer-b0-finetuned-ade-512-512",
   "architectures": [
     "SegformerForSemanticSegmentation"
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
-  "decoder_hidden_size": 768,
+  "decoder_hidden_size": 256,
   "depths": [
-    3,
-    6,
-    40,
-    3
+    2,
+    2,
+    2,
+    2
   ],
   "downsampling_rates": [
     1,
@@ -22,20 +22,20 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_sizes": [
+    32,
     64,
-    128,
-    320,
-    512
+    160,
+    256
   ],
   "id2label": {
     "0": "background",
-    "1": "memes_line"
+    "1": "plate_line"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
     "background": 0,
-    "memes_line": 1
+    "plate_line": 1
   },
   "layer_norm_eps": 1e-06,
   "mlp_ratios": [
@@ -74,5 +74,5 @@
     2
   ],
   "torch_dtype": "float32",
-  "transformers_version": "4.29.0"
+  "transformers_version": "4.37.0"
 }

preprocessor_config.json

@@ -17,7 +17,7 @@
   "resample": 2,
   "rescale_factor": 0.00392156862745098,
   "size": {
-    "height": 640,
-    "width": 640
+    "height": 512,
+    "width": 512
   }
 }

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1084b31ac0a2408eeb2c216186f26c4cc9171cebd3b466a8596d6743425c6ce9
-size 339913320
+oid sha256:bb9455e00ffdcb2c233291d3db728d578868a7b49ea3feefdff8b5b97c403336
+size 15132832