Hugging Face's logo

romulolopes
/
segformer-b0-scene-parse-150

Transformers
TensorBoard
Safetensors
segformer
Inference Endpoints
Model card Files Metrics Community
romulolopes commited on
Commit
d086cfd
•
1 Parent(s): 5f6a133

Training in progress, step 5

Browse files
Files changed (4) hide show
  1. config.json +122 -0
  2. model.safetensors +3 -0
  3. runs/Apr20_15-22-03_ed6ac8148567/events.out.tfevents.1713626524.ed6ac8148567.218.0 +3 -0
  4. training_args.bin +3 -0
config.json ADDED
@@ -0,0 +1,122 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b0",
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,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.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
+ "1": "1",
32
+ "2": "2",
33
+ "3": "3",
34
+ "4": "4",
35
+ "5": "5",
36
+ "6": "6",
37
+ "7": "7",
38
+ "8": "8",
39
+ "9": "9",
40
+ "10": "10",
41
+ "11": "11",
42
+ "12": "12",
43
+ "13": "13",
44
+ "14": "14",
45
+ "15": "15",
46
+ "16": "16",
47
+ "17": "17",
48
+ "18": "18",
49
+ "19": "19",
50
+ "20": "20",
51
+ "21": "21",
52
+ "22": "22",
53
+ "23": "23",
54
+ "24": "24"
55
+ },
56
+ "image_size": 224,
57
+ "initializer_range": 0.02,
58
+ "label2id": {
59
+ "1": 1,
60
+ "10": 10,
61
+ "11": 11,
62
+ "12": 12,
63
+ "13": 13,
64
+ "14": 14,
65
+ "15": 15,
66
+ "16": 16,
67
+ "17": 17,
68
+ "18": 18,
69
+ "19": 19,
70
+ "2": 2,
71
+ "20": 20,
72
+ "21": 21,
73
+ "22": 22,
74
+ "23": 23,
75
+ "24": 24,
76
+ "3": 3,
77
+ "4": 4,
78
+ "5": 5,
79
+ "6": 6,
80
+ "7": 7,
81
+ "8": 8,
82
+ "9": 9
83
+ },
84
+ "layer_norm_eps": 1e-06,
85
+ "mlp_ratios": [
86
+ 4,
87
+ 4,
88
+ 4,
89
+ 4
90
+ ],
91
+ "model_type": "segformer",
92
+ "num_attention_heads": [
93
+ 1,
94
+ 2,
95
+ 5,
96
+ 8
97
+ ],
98
+ "num_channels": 3,
99
+ "num_encoder_blocks": 4,
100
+ "patch_sizes": [
101
+ 7,
102
+ 3,
103
+ 3,
104
+ 3
105
+ ],
106
+ "reshape_last_stage": true,
107
+ "semantic_loss_ignore_index": 255,
108
+ "sr_ratios": [
109
+ 8,
110
+ 4,
111
+ 2,
112
+ 1
113
+ ],
114
+ "strides": [
115
+ 4,
116
+ 2,
117
+ 2,
118
+ 2
119
+ ],
120
+ "torch_dtype": "float32",
121
+ "transformers_version": "4.38.2"
122
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49ced0cdde718166f91da71418e2eda8c1e60dfc02f7b1bbf44cfa99cc58ba24
3
+ size 14907400
runs/Apr20_15-22-03_ed6ac8148567/events.out.tfevents.1713626524.ed6ac8148567.218.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b0e32dd95cb30e07c1cde99490e70838776cd4400095415b97923d0ea3fd994
3
+ size 7121
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:766fa03d3b306f3c4f595ccfb3304d0dbdb878333bd5398e0cf714ed5c44a523
3
+ size 4920