format37 commited on
Commit
ee7fb0d
·
verified ·
1 Parent(s): efc8264

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +41 -0
config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "batch_size": 2,
3
+ "gradient_accumulation_steps": 4,
4
+ "epochs": 200,
5
+ "lr": 5e-05,
6
+ "backbone_lr": 1e-05,
7
+ "weight_decay": 0.0001,
8
+ "backbone_type": "resnet50",
9
+ "use_pretrained": true,
10
+ "freeze_backbone_epochs": 5,
11
+ "use_gradient_checkpointing": true,
12
+ "image_size": [
13
+ 1920,
14
+ 1080
15
+ ],
16
+ "num_classes": 11,
17
+ "merge_banknotes": true,
18
+ "train_ratio": 0.85,
19
+ "test_ratio": 0.15,
20
+ "fpn_channels": 256,
21
+ "strides": [
22
+ 8,
23
+ 16,
24
+ 32,
25
+ 64,
26
+ 128
27
+ ],
28
+ "use_native_resolution": true,
29
+ "focal_alpha": 0.5,
30
+ "focal_gamma": 2.5,
31
+ "lambda_reg": 2.0,
32
+ "lambda_ctr": 1.0,
33
+ "nms_threshold": 0.4,
34
+ "score_threshold": 0.35,
35
+ "max_detections": 100,
36
+ "patience": 30,
37
+ "grad_clip": 1.0,
38
+ "warmup_epochs": 10,
39
+ "hf_dataset": "format37/russian-rubles-banknotes",
40
+ "hf_model_repo": "format37/fcos-banknotes-detector"
41
+ }