| base_checkpoint: "./checkpoints/dune_vitbase14_448_paper.pth" | |
| dune_checkpoint: "./checkpoints/dune_vitbase14_448_paper.pth" | |
| realm_checkpoint: "./checkpoints/REALM_final_rank32_0955116.pth" | |
| embedding: | |
| num_bins: 5 | |
| patch_grid: 32 | |
| embed_dim: 768 | |
| base_channels: 64 | |
| num_downsample: 3 | |
| norm: "batch" | |
| use_residual: True | |
| projector: 'dino2reg_vitlarge_14' | |
| pretrained_head: "./checkpoints/heads/segmentation.pth" | |
| head: | |
| type: "segmentation" | |
| in_channels: 1024 | |
| num_classes: 11 |