config/magic_infer_moge.yaml

seed: 0
# config the hydra logger, only in hydra `$` can be decoded as cite
data: ./assets/room
vis_track: false
hydra:
  run:
    dir: .
  output_subdir: null
  job_logging: {}
  hydra_logging: {}
mixed_precision: bf16
visdom:
  viz_ip: "localhost"
  port: 6666
relax_load: false
res_all: 336
# config the ckpt path
# ckpts: "/mnt/bn/xyxdata/home/codes/my_projs/SpaTrack2/checkpoints/new_base.pth"
ckpts: "Yuxihenry/SpatialTracker_Files"
batch_size: 1
input:
  type: image
fps: 1
model_wind_size: 32
model:
  backbone_cfg:      
    ckpt_dir: "checkpoints/model.pt"
  chunk_size: 24        # downsample factor for patchified features
  ckpt_fwd: true
  ft_cfg:
    mode: "fix"
    paras_name:  []
  resolution: 336
  max_len: 512
  Track_cfg:
    base_ckpt: "checkpoints/scaled_offline.pth"
    base:
      stride: 4
      corr_radius: 3
      window_len: 60
    stablizer: True
    mode: "online"
    s_wind: 200
    overlap: 4
  track_num: 0

dist_train:
  num_nodes: 1