{ | |
"hparams": { | |
"accumulate_grad_batches": 1, | |
"bands": [ | |
0, | |
1, | |
2, | |
3, | |
4, | |
5 | |
], | |
"batch_size": 4, | |
"debug": false, | |
"freeze_backbone": true, | |
"img_size": 224, | |
"lr": 0.001, | |
"max_epochs": 1, | |
"num_frames": 3, | |
"patch_size": 16, | |
"subsets": { | |
"train": 4, | |
"val": 2 | |
}, | |
"tiers": { | |
"tier1": { | |
"loss_weight": 1, | |
"num_classes": 6 | |
}, | |
"tier2": { | |
"loss_weight": 1, | |
"num_classes": 17 | |
}, | |
"tier3": { | |
"loss_weight": 1, | |
"num_classes": 49 | |
}, | |
"tier3_refined": { | |
"loss_weight": 1, | |
"num_classes": 49 | |
} | |
} | |
} | |
} |