davanstrien HF Staff commited on
Commit
785c567
·
verified ·
1 Parent(s): d9f2db2

Training in progress, epoch 1

Browse files
config.json ADDED
@@ -0,0 +1,204 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_dropout": 0.0,
3
+ "activation_function": "silu",
4
+ "anchor_image_size": null,
5
+ "architectures": [
6
+ "DFineForObjectDetection"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "auxiliary_loss": true,
10
+ "backbone": null,
11
+ "backbone_config": {
12
+ "depths": [
13
+ 3,
14
+ 4,
15
+ 6,
16
+ 3
17
+ ],
18
+ "downsample_in_bottleneck": false,
19
+ "downsample_in_first_stage": false,
20
+ "embedding_size": 32,
21
+ "hidden_act": "relu",
22
+ "hidden_sizes": [
23
+ 128,
24
+ 256,
25
+ 512,
26
+ 1024
27
+ ],
28
+ "initializer_range": 0.02,
29
+ "layer_type": "basic",
30
+ "model_type": "hgnet_v2",
31
+ "num_channels": 3,
32
+ "out_features": [
33
+ "stage2",
34
+ "stage3",
35
+ "stage4"
36
+ ],
37
+ "out_indices": [
38
+ 2,
39
+ 3,
40
+ 4
41
+ ],
42
+ "stage_downsample": [
43
+ false,
44
+ true,
45
+ true,
46
+ true
47
+ ],
48
+ "stage_in_channels": [
49
+ 16,
50
+ 64,
51
+ 256,
52
+ 512
53
+ ],
54
+ "stage_kernel_size": [
55
+ 3,
56
+ 3,
57
+ 5,
58
+ 5
59
+ ],
60
+ "stage_light_block": [
61
+ false,
62
+ false,
63
+ true,
64
+ true
65
+ ],
66
+ "stage_mid_channels": [
67
+ 16,
68
+ 32,
69
+ 64,
70
+ 128
71
+ ],
72
+ "stage_names": [
73
+ "stem",
74
+ "stage1",
75
+ "stage2",
76
+ "stage3",
77
+ "stage4"
78
+ ],
79
+ "stage_num_blocks": [
80
+ 1,
81
+ 1,
82
+ 2,
83
+ 1
84
+ ],
85
+ "stage_numb_of_layers": [
86
+ 3,
87
+ 3,
88
+ 3,
89
+ 3
90
+ ],
91
+ "stage_out_channels": [
92
+ 64,
93
+ 256,
94
+ 512,
95
+ 1024
96
+ ],
97
+ "stem_channels": [
98
+ 3,
99
+ 16,
100
+ 16
101
+ ],
102
+ "use_learnable_affine_block": true
103
+ },
104
+ "backbone_kwargs": null,
105
+ "batch_norm_eps": 1e-05,
106
+ "box_noise_scale": 1.0,
107
+ "d_model": 256,
108
+ "decoder_activation_function": "relu",
109
+ "decoder_attention_heads": 8,
110
+ "decoder_ffn_dim": 1024,
111
+ "decoder_in_channels": [
112
+ 256,
113
+ 256,
114
+ 256
115
+ ],
116
+ "decoder_layers": 3,
117
+ "decoder_method": "default",
118
+ "decoder_n_points": [
119
+ 3,
120
+ 6,
121
+ 3
122
+ ],
123
+ "decoder_offset_scale": 0.5,
124
+ "depth_mult": 0.34,
125
+ "dropout": 0.0,
126
+ "encode_proj_layers": [
127
+ 2
128
+ ],
129
+ "encoder_activation_function": "gelu",
130
+ "encoder_attention_heads": 8,
131
+ "encoder_ffn_dim": 1024,
132
+ "encoder_hidden_dim": 256,
133
+ "encoder_in_channels": [
134
+ 256,
135
+ 512,
136
+ 1024
137
+ ],
138
+ "encoder_layers": 1,
139
+ "eos_coefficient": 0.0001,
140
+ "eval_idx": -1,
141
+ "eval_size": null,
142
+ "feat_strides": [
143
+ 8,
144
+ 16,
145
+ 32
146
+ ],
147
+ "focal_loss_alpha": 0.75,
148
+ "focal_loss_gamma": 2.0,
149
+ "freeze_backbone_batch_norms": true,
150
+ "hidden_expansion": 0.5,
151
+ "id2label": {
152
+ "0": "0",
153
+ "1": "1",
154
+ "2": "2",
155
+ "3": "3",
156
+ "4": "4",
157
+ "5": "5",
158
+ "6": "6"
159
+ },
160
+ "initializer_bias_prior_prob": null,
161
+ "initializer_range": 0.01,
162
+ "is_encoder_decoder": true,
163
+ "label2id": {
164
+ "0": 0,
165
+ "1": 1,
166
+ "2": 2,
167
+ "3": 3,
168
+ "4": 4,
169
+ "5": 5,
170
+ "6": 6
171
+ },
172
+ "label_noise_ratio": 0.5,
173
+ "layer_norm_eps": 1e-05,
174
+ "layer_scale": 1,
175
+ "learn_initial_query": false,
176
+ "lqe_hidden_dim": 64,
177
+ "lqe_layers": 2,
178
+ "matcher_alpha": 0.25,
179
+ "matcher_bbox_cost": 5.0,
180
+ "matcher_class_cost": 2.0,
181
+ "matcher_gamma": 2.0,
182
+ "matcher_giou_cost": 2.0,
183
+ "max_num_bins": 32,
184
+ "model_type": "d_fine",
185
+ "normalize_before": false,
186
+ "num_denoising": 100,
187
+ "num_feature_levels": 3,
188
+ "num_queries": 300,
189
+ "positional_encoding_temperature": 10000,
190
+ "reg_scale": 4.0,
191
+ "top_prob_values": 4,
192
+ "torch_dtype": "float32",
193
+ "transformers_version": "4.52.0.dev0",
194
+ "up": 0.5,
195
+ "use_focal_loss": true,
196
+ "use_pretrained_backbone": false,
197
+ "use_timm_backbone": false,
198
+ "weight_loss_bbox": 5.0,
199
+ "weight_loss_ddf": 1.5,
200
+ "weight_loss_fgl": 0.15,
201
+ "weight_loss_giou": 2.0,
202
+ "weight_loss_vfl": 1.0,
203
+ "with_box_refine": true
204
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3881c27e7fbcdf6352299aeb8b467380718acf320235874c47018c9f452f293a
3
+ size 41122684
preprocessor_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_annotations": true,
3
+ "do_normalize": false,
4
+ "do_pad": false,
5
+ "do_rescale": true,
6
+ "do_resize": true,
7
+ "format": "coco_detection",
8
+ "image_mean": [
9
+ 0.485,
10
+ 0.456,
11
+ 0.406
12
+ ],
13
+ "image_processor_type": "RTDetrImageProcessor",
14
+ "image_std": [
15
+ 0.229,
16
+ 0.224,
17
+ 0.225
18
+ ],
19
+ "pad_size": null,
20
+ "resample": 2,
21
+ "rescale_factor": 0.00392156862745098,
22
+ "size": {
23
+ "height": 640,
24
+ "width": 640
25
+ }
26
+ }
runs/May08_13-30-14_fe5c9bb24a70/events.out.tfevents.1746711016.fe5c9bb24a70.50320.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cdac3a4b97f61715a4b5c7765d571ad3d2968be1d989ea12991a013c6c416f6
3
+ size 7980
runs/May08_13-53-55_fe5c9bb24a70/events.out.tfevents.1746712437.fe5c9bb24a70.53415.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1581eb2b1bd85468adf46251648548163caee37b265fbaf91da5006b74ee8df7
3
+ size 40998
runs/May08_13-53-55_fe5c9bb24a70/events.out.tfevents.1746716146.fe5c9bb24a70.53415.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54610570d5cdaaa4f37c35784cfe0ae45a0c6d13f57f5001db33d77edc43b419
3
+ size 359
runs/May08_15-22-46_fe5c9bb24a70/events.out.tfevents.1746717769.fe5c9bb24a70.78541.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15c9c60abcb52818ff85534a4f2a11e278f4fe97a1e96ecf7f0150aac2af18df
3
+ size 11216
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79b63faf5db60aaab1a6b7e6ed9e9a1e570c01bf027089d06f7ad193614c1c8e
3
+ size 5777