VisDa/config.yaml at master · flixpar/VisDa · GitHub

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
{
    model: "GCN_RESNEXT",
    optimizer: "Adam",

    max_epochs: 15,
    batch_size: 2,

    lr: 0.00001,
    weight_decay: 0.0005,
    momentum: 0.99,

    K: 11,

    scale_factor: 0.45,
    default_img_size: !!python/tuple [1024, 2048], # [1052, 1914],

    lr_decay: True,
    scheduler: "poly",
    lr_decay_freq: 1,
    lr_decay_power: 0.85,

    eval_freq: 5000,
    eval_samples: 25,

    resume: False,
    resume_epoch: 0,

    paths: {
        project_path: "/home/flixpar/VisDa",
        data_base_path: "/media/data/",
        data_train_path: "/media/data/train/",
        cityscapes_path: "/media/data/cityscapes/",
        cityscapes_coarse_path: "/media/data/cityscapes_coarse/",
        cityscapes_val_path: "/media/data/cityscapes_eval/",
        pretrained_models_path: "/media/data/pretrained/"
    }

}