osmanh commited on
Commit
0b7646d
1 Parent(s): f5bcf7a

Training in progress, step 20

Browse files
Files changed (4) hide show
  1. config.json +110 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +23 -0
  4. training_args.bin +3 -0
config.json ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "0": "Background",
32
+ "1": "Hat",
33
+ "2": "Hair",
34
+ "3": "Sunglasses",
35
+ "4": "Upper-clothes",
36
+ "5": "Skirt",
37
+ "6": "Pants",
38
+ "7": "Dress",
39
+ "8": "Belt",
40
+ "9": "Left-shoe",
41
+ "10": "Right-shoe",
42
+ "11": "Face",
43
+ "12": "Left-leg",
44
+ "13": "Right-leg",
45
+ "14": "Left-arm",
46
+ "15": "Right-arm",
47
+ "16": "Bag",
48
+ "17": "Scarf"
49
+ },
50
+ "image_size": 224,
51
+ "initializer_range": 0.02,
52
+ "label2id": {
53
+ "Background": 0,
54
+ "Bag": 16,
55
+ "Belt": 8,
56
+ "Dress": 7,
57
+ "Face": 11,
58
+ "Hair": 2,
59
+ "Hat": 1,
60
+ "Left-arm": 14,
61
+ "Left-leg": 12,
62
+ "Left-shoe": 9,
63
+ "Pants": 6,
64
+ "Right-arm": 15,
65
+ "Right-leg": 13,
66
+ "Right-shoe": 10,
67
+ "Scarf": 17,
68
+ "Skirt": 5,
69
+ "Sunglasses": 3,
70
+ "Upper-clothes": 4
71
+ },
72
+ "layer_norm_eps": 1e-06,
73
+ "mlp_ratios": [
74
+ 4,
75
+ 4,
76
+ 4,
77
+ 4
78
+ ],
79
+ "model_type": "segformer",
80
+ "num_attention_heads": [
81
+ 1,
82
+ 2,
83
+ 5,
84
+ 8
85
+ ],
86
+ "num_channels": 3,
87
+ "num_encoder_blocks": 4,
88
+ "patch_sizes": [
89
+ 7,
90
+ 3,
91
+ 3,
92
+ 3
93
+ ],
94
+ "reshape_last_stage": true,
95
+ "semantic_loss_ignore_index": 255,
96
+ "sr_ratios": [
97
+ 8,
98
+ 4,
99
+ 2,
100
+ 1
101
+ ],
102
+ "strides": [
103
+ 4,
104
+ 2,
105
+ 2,
106
+ 2
107
+ ],
108
+ "torch_dtype": "float32",
109
+ "transformers_version": "4.46.3"
110
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:057a1956d963c34f49180fb858a9a71a3fa34e47fa65eb2b1170e33565e49a6b
3
+ size 14901232
preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_reduce_labels": false,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
+ ],
11
+ "image_processor_type": "SegformerImageProcessor",
12
+ "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
+ ],
17
+ "resample": 2,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 512,
21
+ "width": 512
22
+ }
23
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e5f6e5f084d897a0078191bf8f14b4d4f67b28e003ce62f3cac612b04185b7c
3
+ size 5304