aashraychegu commited on
Commit
90de770
·
1 Parent(s): 9904b3b
Files changed (2) hide show
  1. config.json +72 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "SegformerForSemanticSegmentation"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "classifier_dropout_prob": 0.1,
7
+ "decoder_hidden_size": 768,
8
+ "depths": [
9
+ 2,
10
+ 3,
11
+ 4,
12
+ 3
13
+ ],
14
+ "drop_path_rate": 0.1,
15
+ "hidden_act": "gelu",
16
+ "hidden_dropout_prob": 0.0,
17
+ "hidden_sizes": [
18
+ 64,
19
+ 128,
20
+ 320,
21
+ 512
22
+ ],
23
+ "id2label": {
24
+ "0": "sky",
25
+ "1": "surface-to-bed",
26
+ "2": "bed-to-bottom"
27
+ },
28
+ "initializer_range": 0.02,
29
+ "label2id": {
30
+ "bed-to-bottom": 2,
31
+ "sky": 0,
32
+ "surface-to-bed": 1
33
+ },
34
+ "layer_norm_eps": 1e-06,
35
+ "mlp_ratios": [
36
+ 4,
37
+ 4,
38
+ 4,
39
+ 4
40
+ ],
41
+ "model_type": "segformer",
42
+ "num_attention_heads": [
43
+ 1,
44
+ 2,
45
+ 5,
46
+ 8
47
+ ],
48
+ "num_channels": 3,
49
+ "num_encoder_blocks": 4,
50
+ "patch_sizes": [
51
+ 7,
52
+ 3,
53
+ 3,
54
+ 3
55
+ ],
56
+ "reshape_last_stage": true,
57
+ "semantic_loss_ignore_index": 255,
58
+ "sr_ratios": [
59
+ 8,
60
+ 4,
61
+ 2,
62
+ 1
63
+ ],
64
+ "strides": [
65
+ 4,
66
+ 2,
67
+ 2,
68
+ 2
69
+ ],
70
+ "torch_dtype": "float32",
71
+ "transformers_version": "4.31.0"
72
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0f57603d71252b0e7e67d62d521241c94e8db38aa1e44b97a07233f51c13bb0
3
+ size 93128541