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