File size: 482 Bytes
600e8c3 f640e20 600e8c3 f640e20 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | 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 |