config.json

{
  "arch": "unet",
  "encoder_name": "se_resnet50",
  "lr": 0.0001,
  "w_gap": 0.05,
  "w_smooth": 0.05,
  "weight_decay": 1e-05
}