-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcfg.json
64 lines (64 loc) · 1.34 KB
/
cfg.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
{
"backbone": "gcn_resnest101",
"xray_type": "chex",
"gray": false,
"use_se": false,
"img_size": 256,
"crop_size": 224,
"pixel_mean": 128.0,
"pixel_std": 64.0,
"interpolation": 1,
"n_crops": 0,
"imagenet": true,
"crop": false,
"extract_fields": "0,1,2,3,4,5",
"upsample_index": [0,1,2,3,4,5],
"upsample_times": 1,
"offset": 1,
"no_jsd": true,
"augmix": false,
"mish": false,
"wildcat": false,
"bb_freeze": false,
"num_maps": 6,
"num_gcn_maps": 1024,
"beta_mish": 1.0,
"train_name": "train_sss",
"valid_name": "valid_sss",
"source_name": "data",
"csv_path": "/u01/data/Vimmec_Data_small_v4/train_sss.csv",
"zero_limits": 200000,
"ones_limits": 200000,
"agree_rate": 0.5,
"transpose": false,
"embedded_gcn": true,
"within_kernel": true,
"embedded_size": 100,
"train_batch_size": 28,
"dev_batch_size": 28,
"pretrained": true,
"norm_type": "BatchNorm",
"global_pool": "GMP",
"fc_bn": true,
"conv_fc": true,
"use_mish": false,
"attention_map": "None",
"lse_gamma": 10.0,
"fc_drop": 0.0,
"lamb": 1,
"optim": "adam",
"criterion": "bce",
"lr_scheduler": "step",
"t_max": 1,
"lr": 1e-4,
"eta_min": 1e-5,
"lr_factor": 0.1,
"step_size": 2,
"momentum": 0.9,
"weight_decay": 0,
"beta1": 0.9,
"beta2": 0.999,
"threshold": 0.5,
"metric_type": "weighted",
"beta": 1
}