Struct Sam2MemoryConfig 

pub struct Sam2MemoryConfig {

    pub d_model: usize,
    pub num_layers: usize,
    pub num_heads: usize,
    pub dim_feedforward: usize,
    pub layer_norm_eps: f64,
    pub kv_in_dim: usize,
    pub rope_theta: f32,
    pub rope_feat_size: [usize; 2],
    pub rope_k_repeat: bool,
    pub pos_enc_at_input: bool,
    pub pos_enc_at_attn: bool,
    pub pos_enc_at_cross_attn_keys: bool,
    pub pos_enc_at_cross_attn_queries: bool,
    pub max_obj_ptrs_in_encoder: usize,
    pub mem_attn_in_graph_rope: bool,
    pub mem_dim: usize,
}

Memory-attention configuration (video path).

d_model (256) is the working dim of memory-attention layers; memory bank tokens come in at kv_in_dim=out_dim of the memory encoder (64 in published configs) and are projected up by the cross-attention’s k/v linear layers.

Fields

d_model: usize

num_layers: usize

num_heads: usize

dim_feedforward: usize

layer_norm_eps: f64

kv_in_dim: usize

Memory bank kv channel dim — matches memory-encoder out_dim.

rope_theta: f32

rope_feat_size: [usize; 2]

Spatial size used to build the RoPE table for the image features (32×32 for the stride-32 path in published configs).

rope_k_repeat: bool

pos_enc_at_input: bool

Whether to add the input PE to the queries at layer input (pos_enc_at_input in the reference YAML; true).

pos_enc_at_attn: bool

pos_enc_at_cross_attn_keys: bool

pos_enc_at_cross_attn_queries: bool

max_obj_ptrs_in_encoder: usize

Maximum number of object pointers preserved across frames.

mem_attn_in_graph_rope: bool

Fuse each memory-attention layer into one graph with [Op::AxialRope2d] (faster than the default five-graph + host-RoPE path). Requires axial RoPE on the compile device (CPU today; Metal uses host fallback when enabled).

mem_dim: usize

Number of temporal-position embeddings packed with each object-pointer token. Reference YAML: 64.

Trait Implementations

impl Clone for Sam2MemoryConfig

fn clone(&self) -> Sam2MemoryConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Sam2MemoryConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Default for Sam2MemoryConfig

fn default() -> Sam2MemoryConfig

Returns the “default value” for a type.