Spaces:
Running
on
Zero
Running
on
Zero
File size: 972 Bytes
c8d9d42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
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 |