423A35C7/sam2
1
0
Fork 0
mirror of https://github.com/facebookresearch/sam2.git synced 2025-09-18 12:42:48 +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
Show Stats 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: self_attention:
_target_: sam2.modeling.sam.transformer.RoPEAttention _target_: sam2.modeling.sam.transformer.RoPEAttention
rope_theta: 10000.0 rope_theta: 10000.0
feat_sizes: [32, 32] feat_sizes: [64, 64]
embedding_dim: 256 embedding_dim: 256
num_heads: 1 num_heads: 1
downsample_rate: 1 downsample_rate: 1
@ -108,7 +108,7 @@ trainer:
cross_attention: cross_attention:
_target_: sam2.modeling.sam.transformer.RoPEAttention _target_: sam2.modeling.sam.transformer.RoPEAttention
rope_theta: 10000.0 rope_theta: 10000.0
feat_sizes: [32, 32] feat_sizes: [64, 64]
rope_k_repeat: True rope_k_repeat: True
embedding_dim: 256 embedding_dim: 256
num_heads: 1 num_heads: 1