SkalskiP's picture
drop ZERO GPU
44b03d2
raw
history blame
308 Bytes
CHECKPOINTS = {
"tiny": ["sam2_hiera_t.yaml", "checkpoints/sam2_hiera_tiny.pt"],
"small": ["sam2_hiera_s.yaml", "checkpoints/sam2_hiera_small.pt"],
"base_plus": ["sam2_hiera_b+.yaml", "checkpoints/sam2_hiera_base_plus.pt"],
"large": ["sam2_hiera_l.yaml", "checkpoints/sam2_hiera_large.pt"],
}