datvo06 commited on
Commit
68df8ae
1 Parent(s): f991682

Upload tabletransformer/structure_config.json with huggingface_hub

Browse files
tabletransformer/structure_config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "lr":5e-5,
3
+ "lr_backbone":1e-5,
4
+ "batch_size":2,
5
+ "weight_decay":1e-4,
6
+ "epochs":20,
7
+ "lr_drop":1,
8
+ "lr_gamma":0.9,
9
+ "clip_max_norm":0.1,
10
+
11
+ "backbone":"resnet18",
12
+ "num_classes":6,
13
+ "dilation":false,
14
+ "position_embedding":"sine",
15
+ "emphasized_weights":{},
16
+
17
+ "enc_layers":6,
18
+ "dec_layers":6,
19
+ "dim_feedforward":2048,
20
+ "hidden_dim":256,
21
+ "dropout":0.1,
22
+ "nheads":8,
23
+ "num_queries":125,
24
+ "pre_norm":true,
25
+
26
+ "masks":false,
27
+
28
+ "aux_loss":false,
29
+
30
+ "mask_loss_coef":1,
31
+ "dice_loss_coef":1,
32
+ "ce_loss_coef":1,
33
+ "bbox_loss_coef":5,
34
+ "giou_loss_coef":2,
35
+ "eos_coef":0.4,
36
+
37
+ "set_cost_class":1,
38
+ "set_cost_bbox":5,
39
+ "set_cost_giou":2,
40
+
41
+ "device":"cuda",
42
+ "seed":42,
43
+ "start_epoch":0,
44
+ "num_workers":1
45
+ }