davidhajdu commited on
Commit
2f106bf
1 Parent(s): 5a1b388

Training in progress, epoch 1

Browse files
config.json ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "davidhajdu/dab-detr-resnet-50",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "prelu",
5
+ "architectures": [
6
+ "DABDETRForObjectDetection"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "auxiliary_loss": false,
10
+ "backbone": "resnet50",
11
+ "backbone_config": null,
12
+ "backbone_kwargs": {
13
+ "in_chans": 3,
14
+ "out_indices": [
15
+ 1,
16
+ 2,
17
+ 3,
18
+ 4
19
+ ]
20
+ },
21
+ "bbox_cost": 5,
22
+ "bbox_embed_diff_each_layer": false,
23
+ "bbox_loss_coefficient": 5,
24
+ "class_cost": 2,
25
+ "cls_loss_coefficient": 2,
26
+ "d_model": 256,
27
+ "decoder_attention_heads": 8,
28
+ "decoder_bbox_embed_diff_each_layer": false,
29
+ "decoder_ffn_dim": 2048,
30
+ "decoder_layerdrop": 0.0,
31
+ "decoder_layers": 6,
32
+ "decoder_modulate_hw_attn": true,
33
+ "do_use_self_attn_decoder": true,
34
+ "dropout": 0.1,
35
+ "encoder_attention_heads": 8,
36
+ "encoder_ffn_dim": 2048,
37
+ "encoder_layerdrop": 0.0,
38
+ "encoder_layers": 6,
39
+ "focal_alpha": 0.25,
40
+ "giou_cost": 2,
41
+ "giou_loss_coefficient": 2,
42
+ "id2label": {
43
+ "0": "Coverall",
44
+ "1": "Face_Shield",
45
+ "2": "Gloves",
46
+ "3": "Goggles",
47
+ "4": "Mask"
48
+ },
49
+ "init_std": 0.02,
50
+ "init_xavier_std": 1.0,
51
+ "initializer_bias_prior_prob": null,
52
+ "is_encoder_decoder": true,
53
+ "iter_update": true,
54
+ "keep_query_pos": false,
55
+ "label2id": {
56
+ "Coverall": 0,
57
+ "Face_Shield": 1,
58
+ "Gloves": 2,
59
+ "Goggles": 3,
60
+ "Mask": 4
61
+ },
62
+ "model_type": "dab-detr",
63
+ "normalize_before": false,
64
+ "num_hidden_layers": 6,
65
+ "num_patterns": 0,
66
+ "num_queries": 300,
67
+ "position_embedding_type": "sine",
68
+ "query_dim": 4,
69
+ "query_scale_type": "cond_elewise",
70
+ "random_refpoints_xy": false,
71
+ "sine_position_embedding_normalize": true,
72
+ "sine_position_embedding_scale": null,
73
+ "temperature_height": 20,
74
+ "temperature_width": 20,
75
+ "torch_dtype": "float32",
76
+ "transformers_version": "4.46.0.dev0",
77
+ "use_pretrained_backbone": true,
78
+ "use_timm_backbone": true
79
+ }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:716f5f4be8f7638925c66c1a3546a8fb9e5587f98350f2fb5397295f742047e7
3
- size 175252330
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69c4ff554a9460a2cb7108e04535c56ba89c4ad0563735e7211c1bb00dc255c9
3
+ size 175237322
runs/Oct03_12-36-14_224a88b9bb68/events.out.tfevents.1727958976.224a88b9bb68.277.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:625699055234f31723c548cf54280a554c621f242b786815dbdd3c94c7943f32
3
+ size 7647
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f0fa7f8d78c5e70ef898e03b87d92984cbd1d1ee5e730f053c0e96e33aee6b3c
3
- size 5112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71bf8d99eabb027dd73f99a8ef14efd3ea910c17b431d0c4617d280648d6413a
3
+ size 5240