423A35C7/sam2
1
0
Fork 0
mirror of https://github.com/facebookresearch/sam2.git synced 2025-12-27 17:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

update training config as well to be consistent

Browse Source
This commit is contained in:
Chay Ryali
2024-12-11 07:07:23 +00:00
parent beacd9a521
commit a5dc1d5924
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sam2/configs/sam2.1_training/sam2.1_hiera_b+_MOSE_finetune.yaml
View File

@@ -97,7 +97,7 @@ trainer:
self_attention:
_target_: sam2.modeling.sam.transformer.RoPEAttention
rope_theta: 10000.0
feat_sizes: [32, 32]
feat_sizes: [64, 64]
embedding_dim: 256
num_heads: 1
downsample_rate: 1
@@ -108,7 +108,7 @@ trainer:
cross_attention:
_target_: sam2.modeling.sam.transformer.RoPEAttention
rope_theta: 10000.0
feat_sizes: [32, 32]
feat_sizes: [64, 64]
rope_k_repeat: True
embedding_dim: 256
num_heads: 1