MixFairFace/config.yaml at main · fuenwang/MixFairFace · 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
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
dataset_args:
    train:
        mode: 'train'
        shape: &shape [112,112]
        dataset_type: 'BalancedFaceDataset'
        zip_path: ##### The location of your TrainData.zip ######
        augmentation:
            ToTensor: NULL
            #ColorJitter: [0.5, 0.5, 0.5, 0]
            #RandomCrop: [[112, 96]]
            RandomHorizontalFlip: NULL
        loader_args:
            batch_size: 128
            num_workers: &num_workers 24
            shuffle: True
            drop_last: True
            pin_memory: True
    val:
        shape: *shape
        dataset_type: 'RFWDataset_bin'
        dataset_path: #### The folder path of the four bin files (ex. ./RFW_bin/) #######
        augmentation:
            ToTensor: NULL
            #CenterCrop: [[112, 96]]
        loader_args:
            batch_size: 64
            num_workers: *num_workers
            shuffle: False
            drop_last: False
            pin_memory: True
    val-instance:
        shape: *shape
        dataset_type: 'RFWDatasetAll'
        dataset_path: ##### The location of your ValInstData.zip ########
        augmentation:
            ToTensor: NULL
        loader_args:
            batch_size: 64
            num_workers: *num_workers
            shuffle: False
            drop_last: False
            pin_memory: True
learning_args:
    optimizer_args:
        type: 'SGD'
        args:
            lr: 0.1
            momentum: 0.9
            weight_decay: 0.0005
    scheduler_args:
        type: 'MultiStepLR'
        args:
            milestones: [8, 18, 30, 34]
            verbose: True
            gamma: 0.1
network_args:
    save_path: './save'
    encoder_args: NULL
    head_args:
        in_features: 512
        out_features: NULL
        s: 64.0
        m: 0.35

#criterion_args:
#type: 'FocalLoss'
#args:
#    gamma: 2

exp_args:
    seed: 229
    epoch: 40
    device: 'cuda:0'
    exp_path: './experiment'
    exp_freq: 20
    feature_path: './RFW_features'
    num_nodes: 1
    devices: 4