Struct LayerRecord 

pub struct LayerRecord<M>
where
    M: Module + ModuleDisplay + Module,
{
    pub norm: <RmsNorm as Module>::Record,
    pub mamba_block: <M as Module>::Record,
    pub class_latents: EmptyRecord,
    pub class_latents_emb: <Option<Param<Tensor<2>>> as Module>::Record,
}

The record type for the module.

Fields

norm: <RmsNorm as Module>::Record

The module record associative type.

mamba_block: <M as Module>::Record

The module record associative type.

class_latents: EmptyRecord

class_latents_emb: <Option<Param<Tensor<2>>> as Module>::Record

The module record associative type.

Trait Implementations

impl<M> Record for LayerRecord<M>

where M: Module + ModuleDisplay + Module,

type Item<S: PrecisionSettings> = LayerRecordItem<M, S>

Type of the item that can be serialized and deserialized.

fn into_item<S: PrecisionSettings>(self) -> Self::Item<S>

Convert the current record into the corresponding item that follows the given settings.

fn from_item<S: PrecisionSettings>(item: Self::Item<S>, device: &Device) -> Self

Convert the given item into a record.