shubhamWi91 commited on
Commit
06eb0b8
1 Parent(s): 1936ec1

Training in progress, epoch 1

Browse files
config.json CHANGED
@@ -1,29 +1,68 @@
1
  {
2
- "_name_or_path": "SenseTime/deformable-detr",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
6
- "DeformableDetrForObjectDetection"
7
  ],
 
8
  "attention_dropout": 0.0,
9
  "auxiliary_loss": false,
10
- "backbone": "resnet50",
11
- "backbone_config": null,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  "bbox_cost": 5,
13
  "bbox_loss_coefficient": 5,
14
  "class_cost": 1,
15
  "d_model": 256,
16
  "decoder_attention_heads": 8,
17
- "decoder_ffn_dim": 1024,
18
  "decoder_layerdrop": 0.0,
19
  "decoder_layers": 6,
20
  "decoder_n_points": 4,
21
  "dice_loss_coefficient": 1,
22
- "dilation": false,
23
- "disable_custom_kernels": false,
24
  "dropout": 0.1,
25
  "encoder_attention_heads": 8,
26
- "encoder_ffn_dim": 1024,
27
  "encoder_layerdrop": 0.0,
28
  "encoder_layers": 6,
29
  "encoder_n_points": 4,
@@ -69,17 +108,14 @@
69
  "tetrapack": 13
70
  },
71
  "mask_loss_coefficient": 1,
72
- "max_position_embeddings": 1024,
73
- "model_type": "deformable_detr",
74
- "num_channels": 3,
75
- "num_feature_levels": 4,
76
- "num_queries": 300,
77
  "position_embedding_type": "sine",
78
  "torch_dtype": "float32",
79
  "transformers_version": "4.32.1",
80
- "two_stage": false,
81
  "two_stage_num_proposals": 300,
82
- "use_pretrained_backbone": true,
83
- "use_timm_backbone": true,
84
- "with_box_refine": false
85
  }
 
1
  {
2
+ "_name_or_path": "jozhang97/deta-swin-large-o365",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
6
+ "DetaForObjectDetection"
7
  ],
8
+ "assign_first_stage": true,
9
  "attention_dropout": 0.0,
10
  "auxiliary_loss": false,
11
+ "backbone_config": {
12
+ "attention_probs_dropout_prob": 0.0,
13
+ "depths": [
14
+ 2,
15
+ 2,
16
+ 18,
17
+ 2
18
+ ],
19
+ "drop_path_rate": 0.1,
20
+ "embed_dim": 192,
21
+ "encoder_stride": 32,
22
+ "hidden_act": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_size": 1536,
25
+ "image_size": 224,
26
+ "initializer_range": 0.02,
27
+ "layer_norm_eps": 1e-05,
28
+ "mlp_ratio": 4.0,
29
+ "model_type": "swin",
30
+ "num_heads": [
31
+ 6,
32
+ 12,
33
+ 24,
34
+ 48
35
+ ],
36
+ "num_layers": 4,
37
+ "out_features": [
38
+ "stage2",
39
+ "stage3",
40
+ "stage4"
41
+ ],
42
+ "out_indices": [
43
+ 2,
44
+ 3,
45
+ 4
46
+ ],
47
+ "patch_size": 4,
48
+ "path_norm": true,
49
+ "qkv_bias": true,
50
+ "use_absolute_embeddings": false,
51
+ "window_size": 12
52
+ },
53
  "bbox_cost": 5,
54
  "bbox_loss_coefficient": 5,
55
  "class_cost": 1,
56
  "d_model": 256,
57
  "decoder_attention_heads": 8,
58
+ "decoder_ffn_dim": 2048,
59
  "decoder_layerdrop": 0.0,
60
  "decoder_layers": 6,
61
  "decoder_n_points": 4,
62
  "dice_loss_coefficient": 1,
 
 
63
  "dropout": 0.1,
64
  "encoder_attention_heads": 8,
65
+ "encoder_ffn_dim": 2048,
66
  "encoder_layerdrop": 0.0,
67
  "encoder_layers": 6,
68
  "encoder_n_points": 4,
 
108
  "tetrapack": 13
109
  },
110
  "mask_loss_coefficient": 1,
111
+ "max_position_embeddings": 2048,
112
+ "model_type": "deta",
113
+ "num_feature_levels": 5,
114
+ "num_queries": 900,
 
115
  "position_embedding_type": "sine",
116
  "torch_dtype": "float32",
117
  "transformers_version": "4.32.1",
118
+ "two_stage": true,
119
  "two_stage_num_proposals": 300,
120
+ "with_box_refine": true
 
 
121
  }
preprocessor_config.json CHANGED
@@ -9,7 +9,7 @@
9
  0.456,
10
  0.406
11
  ],
12
- "image_processor_type": "DeformableDetrImageProcessor",
13
  "image_std": [
14
  0.229,
15
  0.224,
 
9
  0.456,
10
  0.406
11
  ],
12
+ "image_processor_type": "DetaImageProcessor",
13
  "image_std": [
14
  0.229,
15
  0.224,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:69381d284cf3e2b05cfc00474d70b5908b828985f1d7f65dd088895cebabf818
3
- size 160818833
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16d32a857be405e5d0ec2431472be5ae8fd8278bb09a57a523fc9342f9b502a3
3
+ size 878823725
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e089b15377bd4ec5d4b4ebca9ba6ab6345c4d39f32cd696b4b4d6217a6c93d5f
3
- size 4091
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15a8f17d6f9900b4198920113e77c1faed4705d5ce3d6dd0a4928158d4d9f1fc
3
+ size 4219