config_mnist.json

{
    "name": "mnist_resnet18",
    "seed":123,
    "feature_noise_rate":0.95,

    "arch": {
        "type": "resnet18gray",
        "args": {"num_classes":10}
    },

    "num_classes": 10,

    "data_loader": {
        "type": "MNISTDataLoader",
        "args":{
            "data_dir": "./dir/to/data",
            "batch_size": 128,
            "num_batches": 0,
            "shuffle": true,
            "validation_split": 0.1,
            "num_workers": 0,
            "pin_memory": true
        }
    },


    "optimizer": {
        "type": "SGD",
        "args":{
            "lr": 0.02,
            "momentum": 0.9,
            "weight_decay": 5e-3
        }
    },

    "train_loss": "cross_entropy",
    "val_loss": "cross_entropy",
    "metrics": [
        "my_metric", "my_metric2"
    ],

    "lr_scheduler": {
        "type": "MultiStepLR",
        "args": { 
            "milestones": [100],
            "gamma": 0.1
        }
    },

    "trainer": {
        "epochs": 200,
        "warmup": 0,
        "save_dir": "./dir/to/model",
        "save_period": 100,
        "verbosity": 2,
        "label_dir": "saved/",
        
        "monitor": "min loss",
        "early_stop": 4000,

        "tensorboard": false,
        "mlflow": false,

        "_percent": "Percentage of noise",
        "percent": 0.8,
        "_asym": "symmetric noise if false",
        "asym": false
    }
}