pj2/resnet.json at main · UntitledNotebook/pj2 · 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
{
    "name": "ResNet",
    "n_gpu": 1,

    "model": "resnet",
    "weights": "./saved/models/ResNet/0313_110036/model_best.pth",

    "train_data_loader": {
        "type": "CIFAR10_4x_DataLoader",
        "args":{
            "root": "data/",
            "split": "train",
            "batch_size": 32,
            "shuffle": true,
            "num_workers": 2
        }
    },
    "valid_data_loader": {
        "type": "CIFAR10_4x_DataLoader",
        "args":{
            "root": "data/",
            "split": "valid",
            "batch_size": 128,
            "shuffle": false,
            "num_workers": 2
        }
    },
    "optimizer": {
        "type": "AdamW",
        "args":{
            "lr": 1e-2,
            "weight_decay": 1e-4
        }
    },
    "loss": {
        "type": "CrossEntropyLoss",
        "args": {
            "label_smoothing": 0.1
        }
    },
    "metrics": [
        "accuracy"
    ],
    "lr_scheduler": {
        "type": "StepLR",
        "args": {
            "step_size": 100,
            "gamma": 0.1
        }
    },
    "train_transform": [
        {
            "type": "AutoAugment",
            "args": {
                "type": "cifar10"
            }
        }
    ],
    "valid_transform": [],
    "mixup": true,
    "mix_precision": true,
    "gradient_clipping": 2.0,
    "csv": true,
    "trainer": {
        "epochs": 200,

        "save_dir": "saved/",
        "save_period": 1,
        "verbosity": 2,

        "monitor": "max val_accuracy",
        "early_stop": 300,

        "tensorboard": false
    }
}