File size: 2,125 Bytes
d4e325a a11e38c 27f6e01 d4e325a a11e38c 323edef c54ab0c 860e12e e9a5d3a 47d9ca4 a11e38c d4e325a 2a9c830 a11e38c 070de2c a11e38c 860e12e a11e38c 2a9c830 d4e325a a11e38c 070de2c a11e38c 860e12e a11e38c e0a9c20 d4e325a 860e12e a11e38c 860e12e a11e38c 860e12e a11e38c 860e12e d4e325a 860e12e d4e325a a11e38c d4e325a a11e38c d4e325a 2155ee4 f5c4ce5 a149248 af02dc8 d4e325a a11e38c d4e325a 2155ee4 d4e325a 8a5aa02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
{
"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
]
}
} |