# DA3-small variant of ZipSplatRefine.
# Inherits the standard refine config, swaps the base backbone to DA3-small
# and points base_weights at the pretrained zipsplat-v4-da3s checkpoint.
# dim_z drops from 1664 (Giant) → 512 (Small), so the refiner's hidden_dim
# is no longer a tight bottleneck. Memory and wall time also drop substantially,
# so we can push BS up.
defaults:
  - model: zipsplat_refine
  - data: composed_252
  - train: zipsplat_refine
  - benchmarks: re10k_dl3dv
  - override /model/backbone@model.base.backbone: da3s
  - _self_

model:
  base_weights: /mnt/hdd/splatfactory_outputs/outputs/training/zipsplat-v4-da3s/checkpoint_best.tar
  base:
    query_sample_ratio: [1.0, 1.0]
    query_ratio_schedule: 0.0
    query_scale_with_views: 0.0
    return_attention: false
  # With dim_z=512 we can match it without breaking the bank.
  refiner:
    hidden_dim: 512
    num_layers: 2
    num_heads: 8
  K: 2

data:
  batch_size: 18
  image_num_range: [2, 18]
  view_schedule: 0.0
  num_target_views: 4
  preprocessing:
    resize: 252
    edge_divisible_by: 14

train:
  seed: 1
  num_steps: 30_000
  lr: 3e-4
