REALM / configs /depth.yaml
viciopoli's picture
Update configs/depth.yaml
c978756 verified
raw
history blame contribute delete
461 Bytes
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
pretrained_head: "./checkpoints/heads/depth.pth"
head:
type: depth
embed_dim: 768
min_depth: 1.95
max_depth: 82.0
num_bins: 256