Hiep Tran commited on
Commit
accef87
·
verified ·
1 Parent(s): e109e7b

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +31 -0
config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "vtdetr",
3
+ "architectures": ["VTDETRModel"],
4
+ "auto_map": {
5
+ "AutoConfig": "modeling_vtdetr.VTDETRConfig",
6
+ "AutoModel": "modeling_vtdetr.VTDETRModel"
7
+ },
8
+ "num_classes": 4,
9
+ "id2label": {
10
+ "0": "Object-detection",
11
+ "1": "Figure",
12
+ "2": "Icon",
13
+ "3": "Table"
14
+ },
15
+ "label2id": {
16
+ "Object-detection": 0,
17
+ "Figure": 1,
18
+ "Icon": 2,
19
+ "Table": 3
20
+ },
21
+ "resolution": 640,
22
+ "checkpoint": "checkpoint_best_total.pth",
23
+ "num_select": 300,
24
+ "group_detr": 13,
25
+ "multi_scale": true,
26
+ "expanded_scales": true,
27
+ "square_resize_div_64": true,
28
+ "weight_decay": 0.0001,
29
+ "drop_path": 0.0,
30
+ "ema_decay": 0.993
31
+ }