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
 ]
 }
}