config.json

{
  "arch": "unet",
  "encoder": "seresnext50_32x4d",
  "in_chans": 3,
  "log_dice": false,
  "lr": 0.003,
  "n_classes": 5,
  "optimizer": "adamw",
  "pretrained": true,
  "smooth_dice": 0.4,
  "test_dataset_id": [
    "IDRID",
    "FGADR",
    "MESSIDOR",
    "DDR",
    "RETLES"
  ]
}