config.json

{
  "name": "UNet",
  "n_gpu": 1,
  "arch": {
    "type": "UNet",
    "args": {
      "num_classes": 19
    }
  },
  "train_data_loader": {
    "type": "CityScapesDataLoader",
    "args": {
      "data_dir": "CityScapes",
      "phase": "train",
      "batch_size": 8,
      "shuffle": true,
      "num_workers": 8
    }
  },
  "valid_data_loader": {
    "type": "CityScapesDataLoader",
    "args": {
      "data_dir": "CityScapes",
      "phase": "val",
      "batch_size": 4,
      "shuffle": true,
      "num_workers": 8
    }
  },
  "optimizer": {
    "type": "SGD",
    "args": {
      "lr": 0.05,
      "weight_decay": 0.0005,
      "momentum": 0.9
    }
  },
  "loss": {
    "type": "CrossEntropyLoss",
    "args": {

    }
  },
  "metrics": [
    "accuracy",
    "mIoU"
  ],
  "lr_scheduler": {
    "type": "StepLR",
    "args": {
      "step_size": 1,
      "gamma": 0.997
    }
  },
  "trainer": {
    "epochs": 484,
    "save_dir": "saved/",
    "save_period": 30,
    "verbosity": 2,
    "monitor": "min val_loss",
    "early_stop": -1,
    "tensorboard": true
  }
}