{
  "architectures": [
    "Terratorch"
  ],
  "num_classes": 0,
  "pretrained_cfg": {
    "seed_everything": 0,
    "input":{
      "target": "pixel_values",
      "data":{
        "pixel_values":{
          "type": "torch.Tensor",
          "shape": [6, 512, 512]
        },
        "location_coords":{
          "type":"torch.Tensor",
          "shape": [1, 2]
        }
      }
    },
    "model": {
      "class_path": "terratorch.tasks.SemanticSegmentationTask",
      "init_args": {
        "model_args": {
          "backbone_pretrained": true,
          "backbone": "prithvi_eo_v2_300_tl",
          "decoder": "UperNetDecoder",
          "decoder_channels": 256,
          "decoder_scale_modules": true,
          "num_classes": 2,
          "rescale": true,
          "backbone_bands": [
            "BLUE",
            "GREEN",
            "RED",
            "NIR_NARROW",
            "SWIR_1",
            "SWIR_2"
          ],
          "head_dropout": 0.1,
          "necks": [
            {
              "name": "SelectIndices",
              "indices": [
                5,
                11,
                17,
                23
              ]
            },
            {
              "name": "ReshapeTokensToImage"
            }
          ]
        },
        "model_factory": "EncoderDecoderFactory",
        "loss": "ce",
        "ignore_index": -1,
        "lr": 0.001,
        "freeze_backbone": false,
        "freeze_decoder": false,
        "plot_on_val": 10
      }
    },
    "data": {
      "class_path": "terratorch.datamodules.Sen1Floods11NonGeoDataModule",
      "init_args": {
        "data_root": "/dccstor/geofm-finetuning/datasets/sen1floods11",
        "batch_size": 16,
        "num_workers": 8,
        "bands": [
          "BLUE",
          "GREEN",
          "RED",
          "NIR_NARROW",
          "SWIR_1",
          "SWIR_2"
        ],
        "train_transform": [
          {
            "class_path": "albumentations.RandomCrop",
            "init_args": {
              "height": 224,
              "width": 224,
              "p": 1
            }
          },
          {
            "class_path": "albumentations.HorizontalFlip",
            "init_args": {
              "p": 0.5
            }
          },
          {
            "class_path": "albumentations.VerticalFlip",
            "init_args": {
              "p": 0.5
            }
          },
          {
            "class_path": "albumentations.pytorch.ToTensorV2",
            "init_args": {
              "transpose_mask": false,
              "p": 1
            }
          }
        ],
        "val_transform": [
          {
            "class_path": "albumentations.pytorch.ToTensorV2",
            "init_args": {
              "transpose_mask": false,
              "p": 1
            }
          }
        ],
        "test_transform": [
          {
            "class_path": "albumentations.pytorch.ToTensorV2",
            "init_args": {
              "transpose_mask": false,
              "p": 1
            }
          }
        ],
        "drop_last": true,
        "constant_scale": 0.0001,
        "no_data_replace": 0,
        "no_label_replace": -1,
        "use_metadata": false
      }
    }
  }
}