| { |
| "hparams": { |
| "accumulate_grad_batches": 1, |
| "backbone_weights_path": null, |
| "bands": [ |
| 0, |
| 1, |
| 2, |
| 3, |
| 4, |
| 5 |
| ], |
| "batch_size": 16, |
| "debug": false, |
| "dropout_p": 0.1, |
| "early_stopping_metric": "val_f1_tier3_majority", |
| "early_stopping_mode": "max", |
| "early_stopping_patience": 10, |
| "experiment_group": "exp-4-backbone", |
| "experiment_name": "exp-4-backbone-not_pretrained", |
| "freeze_backbone": false, |
| "heads_spec": { |
| "tier1": { |
| "is_metrics_tier": true, |
| "loss_weight": 1, |
| "num_classes_to_predict": 6, |
| "target_idx": 0, |
| "type": "HierarchicalFCNHead" |
| }, |
| "tier2": { |
| "is_metrics_tier": true, |
| "loss_weight": 1, |
| "num_classes_to_predict": 17, |
| "target_idx": 1, |
| "type": "HierarchicalFCNHead" |
| }, |
| "tier3": { |
| "is_last_tier": true, |
| "is_metrics_tier": true, |
| "loss_weight": 1, |
| "num_classes_to_predict": 49, |
| "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, |
| "lr": 0.001, |
| "max_epochs": 400, |
| "name": "not_pretrained", |
| "num_frames": 3, |
| "optimizer": "Adam", |
| "optimizer_momentum": null, |
| "optimizer_weight_decay": null, |
| "patch_size": 16, |
| "test_folds": [ |
| 5 |
| ], |
| "train_folds": [ |
| 0, |
| 1, |
| 2, |
| 4 |
| ], |
| "val_folds": [ |
| 3 |
| ] |
| } |
| } |
