config.json

{
  "hparams": {
    "accumulate_grad_batches": 1,
    "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
    "bands": [
      0,
      1,
      2,
      3,
      4,
      5
    ],
    "batch_size": 4,
    "debug": false,
    "freeze_backbone": false,
    "img_size": 224,
    "lr": 0.001,
    "max_epochs": 1,
    "num_frames": 3,
    "patch_size": 16,
    "tiers": {
      "tier1": {
        "loss_weight": 1,
        "num_classes": 6
      },
      "tier2": {
        "loss_weight": 1,
        "num_classes": 17
      },
      "tier3": {
        "loss_weight": 1,
        "num_classes": 49
      },
      "tier3_refined": {
        "loss_weight": 1,
        "num_classes": 49
      }
    }
  }
}