config_classification.json 1.13 KB
Newer Older
Alessia Marcolini's avatar
Alessia Marcolini committed
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
{
    "train_dataset_params": {
        "dataset_name": "tcga_breast",
        "dataset_path": "",
        "loader_params": {
            "batch_size": 32,
            "shuffle": true,
            "num_workers": 4,
            "pin_memory": true
        },
        "input_size": [200, 200]
    },
    "val_dataset_params": {
        "dataset_name": "tcga_breast",
        "dataset_path": "",
        "loader_params": {
            "batch_size": 24,
            "shuffle": false,
            "num_workers": 4,
            "pin_memory": true
        },
        "input_size": [200, 200]
    },
    "model_params": {
        "model_name": "classificationTile",
        "version" : "152",
        "pretrained_imagenet" : true,
        "n_classes" : 2,
        "dropout" : 0.5,
        "optimizer" : "Adam",
        "is_train": true,
        "max_epochs": 40,
        "lr": 0.01,
        "export_path": "",
        "checkpoint_path": "",
        "load_checkpoint": -1,
        "lr_policy": "step",
        "lr_decay_iters": 10
    },
    "visualization_params": {
        "name": "2d segmentation"
    },
    "printout_freq": 10,
    "model_update_freq": 1
}