facebookresearch / facebookresearch/sam2
Checkpoint Loading Error When Increasing num_maskmem in SAM2
Open
- Dominant language
- Jupyter Notebook
- Stars
- 19.9k
- Forks
- 2.5k
- PR merge metrics
- No merged PRs in 30d
Description
Changing num_maskmem from 7 to 21 in sam2.1_hiera_l.yaml causes a checkpoint loading error due to a size mismatch in maskmem_tpos_enc. The model expects shape [21,1,1,64], but the checkpoint has [7,1,1,64]. Should the model support flexible num_maskmem, or is retraining required for new values?
Contributor guide
Assessment
This issue has not been assessed yet.