-
Notifications
You must be signed in to change notification settings - Fork 93
/
example_AVG.json
43 lines (43 loc) · 1.02 KB
/
example_AVG.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
{
"train_csv": "/Path/to/train.csv",
"dev_csv": "/Path/to/dev.csv",
"backbone": "densenet121",
"width": 512,
"height": 512,
"long_side": 512,
"fix_ratio": true,
"pixel_mean": 128.0,
"pixel_std": 64.0,
"use_pixel_std": true,
"use_equalizeHist": true,
"use_transforms_type": "Aug",
"gaussian_blur": 3,
"border_pad": "pixel_mean",
"num_classes": [1,1,1,1,1],
"batch_weight": true,
"enhance_index": [2,6],
"enhance_times": 1,
"pos_weight": [1,1,1,1,1],
"train_batch_size": 24,
"dev_batch_size": 24,
"pretrained": true,
"log_every": 10,
"test_every": 100,
"epoch": 3,
"norm_type": "BatchNorm",
"global_pool": "AVG",
"fc_bn": false,
"attention_map": "None",
"lse_gamma": 0.5,
"fc_drop": 0,
"optimizer": "Adam",
"criterion": "BCE",
"lr": 0.0001,
"lr_factor": 0.1,
"lr_epochs": [2],
"momentum": 0.9,
"weight_decay": 0.0,
"best_target": "auc",
"save_top_k": 3,
"save_index": [0,1,2,3,4]
}