WillNewbo12 commited on
Commit
924c712
1 Parent(s): dd27de3

Training in progress, step 200

Browse files
config.json CHANGED
@@ -1,61 +1,46 @@
1
  {
2
- "_commit_hash": "480370a8aeeed9fc8d78837b4e94e5f936fe73f2",
3
- "_name_or_path": "facebook/detr-resnet-50",
4
- "activation_dropout": 0.0,
5
- "activation_function": "relu",
6
  "architectures": [
7
- "DetrForObjectDetection"
8
  ],
9
- "attention_dropout": 0.0,
10
  "auxiliary_loss": false,
11
- "backbone": "resnet50",
12
- "backbone_config": null,
13
  "bbox_cost": 5,
14
  "bbox_loss_coefficient": 5,
15
  "class_cost": 1,
16
- "classifier_dropout": 0.0,
17
- "d_model": 256,
18
- "decoder_attention_heads": 8,
19
- "decoder_ffn_dim": 2048,
20
- "decoder_layerdrop": 0.0,
21
- "decoder_layers": 6,
22
- "dice_loss_coefficient": 1,
23
- "dilation": false,
24
- "dropout": 0.1,
25
- "encoder_attention_heads": 8,
26
- "encoder_ffn_dim": 2048,
27
- "encoder_layerdrop": 0.0,
28
- "encoder_layers": 6,
29
  "eos_coefficient": 0.1,
30
  "giou_cost": 2,
31
  "giou_loss_coefficient": 2,
 
 
 
32
  "id2label": {
33
- "0": "Coverall",
34
- "1": "Face_Shield",
35
- "2": "Gloves",
36
- "3": "Goggles",
37
- "4": "Mask"
38
  },
39
- "init_std": 0.02,
40
- "init_xavier_std": 1.0,
41
- "is_encoder_decoder": true,
 
 
 
42
  "label2id": {
43
- "Coverall": 0,
44
- "Face_Shield": 1,
45
- "Gloves": 2,
46
- "Goggles": 3,
47
- "Mask": 4
48
  },
49
- "mask_loss_coefficient": 1,
50
- "max_position_embeddings": 1024,
51
- "model_type": "detr",
52
  "num_channels": 3,
53
- "num_hidden_layers": 6,
54
- "num_queries": 100,
55
- "position_embedding_type": "sine",
56
- "scale_embedding": false,
57
  "torch_dtype": "float32",
58
- "transformers_version": "4.7.0.dev0",
59
- "use_pretrained_backbone": true,
60
- "use_timm_backbone": true
61
  }
 
1
  {
2
+ "_name_or_path": "hustvl/yolos-small",
 
 
 
3
  "architectures": [
4
+ "YolosForObjectDetection"
5
  ],
6
+ "attention_probs_dropout_prob": 0.0,
7
  "auxiliary_loss": false,
 
 
8
  "bbox_cost": 5,
9
  "bbox_loss_coefficient": 5,
10
  "class_cost": 1,
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  "eos_coefficient": 0.1,
12
  "giou_cost": 2,
13
  "giou_loss_coefficient": 2,
14
+ "hidden_act": "gelu",
15
+ "hidden_dropout_prob": 0.0,
16
+ "hidden_size": 384,
17
  "id2label": {
18
+ "0": "Debris-on-Formula-One-Circuits",
19
+ "1": "Debris",
20
+ "2": "Marshal",
21
+ "3": "Recovery Vehicle"
 
22
  },
23
+ "image_size": [
24
+ 512,
25
+ 864
26
+ ],
27
+ "initializer_range": 0.02,
28
+ "intermediate_size": 1536,
29
  "label2id": {
30
+ "Debris": 1,
31
+ "Debris-on-Formula-One-Circuits": 0,
32
+ "Marshal": 2,
33
+ "Recovery Vehicle": 3
 
34
  },
35
+ "layer_norm_eps": 1e-12,
36
+ "model_type": "yolos",
37
+ "num_attention_heads": 6,
38
  "num_channels": 3,
39
+ "num_detection_tokens": 100,
40
+ "num_hidden_layers": 12,
41
+ "patch_size": 16,
42
+ "qkv_bias": true,
43
  "torch_dtype": "float32",
44
+ "transformers_version": "4.27.4",
45
+ "use_mid_position_embeddings": true
 
46
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b94c592e26e8bdc46003015275d6938397e7a8de3ec15dfd90f8850f354cd29
3
- size 166620197
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1719a638b26427897e265fd9d1d6c1efa43a6ae4ff9c44f49088987ca6e7c12d
3
+ size 122677021
runs/Apr04_09-12-17_25606165d062/1680599598.1310017/events.out.tfevents.1680599598.25606165d062.129.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cde74f811ba3c3ebbca9cfa32e6dc75a51d7eef99497c758268204b53057c128
3
+ size 5856
runs/Apr04_09-12-17_25606165d062/1680599652.0156121/events.out.tfevents.1680599652.25606165d062.129.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5258ca14ddb70c3653dbbd9e3574354502d350188ae9426b0c33ac1f171a591
3
+ size 5856
runs/Apr04_09-12-17_25606165d062/events.out.tfevents.1680599598.25606165d062.129.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:292eba1c63d6cb7890f2acbfc4f0086d947c1174597064cc35ef53e8f440e06b
3
+ size 4393
runs/Apr04_09-12-17_25606165d062/events.out.tfevents.1680599652.25606165d062.129.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cdd5b77d7c71abd93c2606565cad5a1a284d4cd25c064052403d091f7265531
3
+ size 5015
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:329080adb1478daed48599cc0f6ee18de3590be1a8f2d62a02321576f42d6e31
3
  size 3579
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f74548cf538041b418bdf8d1e9445f48e7ae6a33fb25b4c0eae04e128e17fef
3
  size 3579