config.json

{
  "architecture": "convnext_small",
  "as_regression": true,
  "ema": {
    "decay": 0.999
  },
  "img_size": [
    512,
    512
  ],
  "lr": 3e-05,
  "num_classes": 5,
  "optimizer": {
    "betas": [
      0.9,
      0.999
    ],
    "weight_decay": 0.0005
  },
  "pretrained": true,
  "swa": {
    "swa_lrs": 3e-05
  }
}