config_unet.json

{
    "name": "JITNet",
    "n_gpu": 4,
    "use_synch_bn": false,

    "arch": {
        "type": "UNet",
        "args": {
            "freeze_bn": false
        }
    },

    "train_loader": {
        "type": "COCO",
        "args":{
            "partition": "CocoStuff164k",
            "data_dir": "/home/cfan/coco2017/",
            "batch_size": 24,
            "base_size": 400,
            "crop_size": 380,
            "augment": true,
            "shuffle": true,
            "scale": true,
            "flip": true,
            "rotate": true,
            "blur": false,
            "split": "train2017",
            "num_workers": 8,
            "things_only": false
        }
    },

    "val_loader": {
        "type": "COCO",
        "args":{
            "partition": "CocoStuff164k",
            "data_dir": "/home/cfan/coco2017",
            "batch_size": 8,
            "crop_size": 400,
            "val": true,
            "split": "val2017",
            "num_workers": 4,
            "things_only": false
        }
    },

    "optimizer": {
        "type": "SGD",
        "differential_lr": false,
        "args":{
            "lr": 0.01,
            "weight_decay": 1e-4,
            "momentum": 0.9
        }
    },

    "loss": "CrossEntropyLoss2d",
    "ignore_index": 255,
    "lr_scheduler": {
        "type": "Poly",
        "args": {
        }
    },

    "trainer": {
        "epochs": 100,
        "save_dir": "saved/",
        "save_period": 1,

        "monitor": "max Mean_IoU",
        "early_stop": 10,

        "tensorboard": true,
        "log_dir": "saved/runs",
        "log_per_iter": 20,

        "val": true,
        "val_per_epochs": 1
    }
}