config.json

{
  "hparams": {
    "accumulate_grad_batches": 2,
    "backbone_weights_path": "huggingface",
    "bands": [
      0,
      1,
      2,
      3,
      4,
      5
    ],
    "batch_size": 8,
    "bottleneck_reduction_factor": 4,
    "data_augmentation": {
      "enabled": true,
      "flip_prob": 0.5,
      "jitter_std": 0.01
    },
    "debug": false,
    "dropout_p": 0.1,
    "early_stopping_metric": "val_f1_tier3_majority",
    "early_stopping_mode": "max",
    "early_stopping_patience": 10,
    "experiment_group": "best-model",
    "experiment_name": "best-model-messis",
    "freeze_backbone": false,
    "heads_spec": {
      "tier1": {
        "is_metrics_tier": true,
        "kernel_size": 1,
        "loss_weight": 1,
        "num_channels": 1024,
        "num_classes_to_predict": 6,
        "num_convs": 1,
        "target_idx": 0,
        "type": "HierarchicalFCNHead"
      },
      "tier2": {
        "is_metrics_tier": true,
        "kernel_size": 1,
        "loss_weight": 1,
        "num_channels": 1024,
        "num_classes_to_predict": 17,
        "num_convs": 1,
        "target_idx": 1,
        "type": "HierarchicalFCNHead"
      },
      "tier3": {
        "is_last_tier": true,
        "is_metrics_tier": true,
        "kernel_size": 1,
        "loss_weight": 1,
        "num_channels": 1024,
        "num_classes_to_predict": 49,
        "num_convs": 1,
        "target_idx": 2,
        "type": "HierarchicalFCNHead"
      },
      "tier3_refinement_head": {
        "is_final_head": true,
        "is_metrics_tier": false,
        "loss_weight": 1,
        "num_classes_to_predict": 49,
        "target_idx": 2,
        "type": "LabelRefinementHead"
      }
    },
    "img_size": 224,
    "loss_ignore_background": true,
    "lr": 0.0001,
    "max_epochs": 400,
    "name": "messis",
    "num_frames": 9,
    "optimizer": "Adam",
    "optimizer_momentum": null,
    "optimizer_weight_decay": null,
    "patch_size": 16,
    "test_folds": [
      5
    ],
    "train_folds": [
      1,
      2,
      3,
      4
    ],
    "use_bottleneck_neck": false,
    "val_folds": [
      0
    ]
  }
}