| { |
| "hparams": { |
| "accumulate_grad_batches": 1, |
| "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt", |
| "bands": [ |
| 0, |
| 1, |
| 2, |
| 3, |
| 4, |
| 5 |
| ], |
| "batch_size": 16, |
| "bottleneck_reduction_factor": 8, |
| "data_augmentation": { |
| "enabled": false, |
| "flip_prob": 0.5, |
| "jitter_std": 0 |
| }, |
| "debug": false, |
| "dropout_p": 0.1, |
| "early_stopping_metric": "val_f1_tier2_majority", |
| "early_stopping_mode": "max", |
| "early_stopping_patience": 10, |
| "experiment_group": "exp-7-label-hierarchy", |
| "experiment_name": "exp-7-label-hierarchy-seasonality", |
| "freeze_backbone": true, |
| "heads_spec": { |
| "tier1": { |
| "is_metrics_tier": true, |
| "loss_weight": 1, |
| "num_classes_to_predict": 4, |
| "target_idx": 0, |
| "type": "HierarchicalFCNHead" |
| }, |
| "tier2": { |
| "is_last_tier": true, |
| "is_metrics_tier": true, |
| "loss_weight": 1, |
| "num_classes_to_predict": 43, |
| "target_idx": 1, |
| "type": "HierarchicalFCNHead" |
| }, |
| "tier2_refinement_head": { |
| "is_final_head": true, |
| "is_metrics_tier": false, |
| "loss_weight": 1, |
| "num_classes_to_predict": 43, |
| "target_idx": 1, |
| "type": "LabelRefinementHead" |
| } |
| }, |
| "img_size": 224, |
| "loss_ignore_background": false, |
| "lr": 0.001, |
| "max_epochs": 400, |
| "name": "seasonality", |
| "num_frames": 3, |
| "optimizer": "Adam", |
| "optimizer_momentum": null, |
| "optimizer_weight_decay": null, |
| "patch_size": 16, |
| "test_folds": [ |
| 5 |
| ], |
| "train_folds": [ |
| 0, |
| 1, |
| 2, |
| 3 |
| ], |
| "use_bottleneck_neck": false, |
| "val_folds": [ |
| 4 |
| ] |
| } |
| } |
