Omar280X commited on
Commit
917649f
·
1 Parent(s): 6f18217

Training in progress steps 4

Browse files
Files changed (2) hide show
  1. config.json +9 -9
  2. tf_model.h5 +2 -2
config.json CHANGED
@@ -1,16 +1,16 @@
1
  {
2
- "_name_or_path": "nvidia/segformer-b2-finetuned-ade-512-512",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
  "classifier_dropout_prob": 0.1,
8
- "decoder_hidden_size": 768,
9
  "depths": [
10
- 3,
11
- 4,
12
- 6,
13
- 3
14
  ],
15
  "downsampling_rates": [
16
  1,
@@ -22,10 +22,10 @@
22
  "hidden_act": "gelu",
23
  "hidden_dropout_prob": 0.0,
24
  "hidden_sizes": [
 
25
  64,
26
- 128,
27
- 320,
28
- 512
29
  ],
30
  "id2label": {
31
  "0": "background",
 
1
  {
2
+ "_name_or_path": "nvidia/segformer-b0-finetuned-ade-512-512",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
  "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 256,
9
  "depths": [
10
+ 2,
11
+ 2,
12
+ 2,
13
+ 2
14
  ],
15
  "downsampling_rates": [
16
  1,
 
22
  "hidden_act": "gelu",
23
  "hidden_dropout_prob": 0.0,
24
  "hidden_sizes": [
25
+ 32,
26
  64,
27
+ 160,
28
+ 256
 
29
  ],
30
  "id2label": {
31
  "0": "background",
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c6b0fc06e09cb3390d6ad99a43ff6bd4d1e360c9156871a4c66382d8bb3951b2
3
- size 109913384
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22d02c072762dbc7de125b9a09d67fbba7817977233ba91f43de682e991a0c59
3
+ size 15133552