configcit_fcn.json

{
    "name": "FCN8",
    "n_gpu": 4,
    "use_synch_bn": true,

    "arch": {
        "type": "FCN8",
        "args": {
            "backbone": "resnet50",
            "freeze_bn": false,
            "freeze_backbone": true
        }
    },

    "train_loader": {
        "type": "CityScapes",
        "args":{
            "data_dir": "Cityscape",
            "batch_size":15,
            "base_size": 1024,
            "crop_size": 768,
            "augment": true,
           
            "scale": true,
            "flip": true,
            "rotate": true,
            "blur": false,
			
            "split": "train",
            "num_workers": 8
        }
    },

    "val_loader": {
        "type": "CityScapes",
        "args":{
            "data_dir": "Cityscape",
            "batch_size": 15,
            "base_size": 512,
            "val": true,
            "augment": false,
            "flip":false,
            "split": "val",
            "num_workers": 6
        }
    },

  
    "optimizer": {
        "type": "SGD",
        "differential_lr": true,
        "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": 3,
  
        "monitor": "max Mean_IoU",
        "early_stop": 10,
        
        "tensorboard": true,
        "log_dir": "saved/runs",
        "log_per_iter": 100,

        "val": true,
        "val_per_epochs": 3
    }
}