Struct VocabNetworkRecord 

pub struct VocabNetworkRecord<M>
where
    M: Module + ModuleDisplay + Module,
{
    pub embedding: <Embedding as Module>::Record,
    pub layers: <Layers<M> as Module>::Record,
    pub norm_f: <RmsNorm as Module>::Record,
    pub lm_head: <Option<Linear> as Module>::Record,
}

The record type for the module.

Fields

embedding: <Embedding as Module>::Record

The module record associative type.

layers: <Layers<M> as Module>::Record

The module record associative type.

norm_f: <RmsNorm as Module>::Record

The module record associative type.

lm_head: <Option<Linear> as Module>::Record

The module record associative type.

Trait Implementations

impl<M> Record for VocabNetworkRecord<M>

where M: Module + ModuleDisplay + Module,

type Item<S: PrecisionSettings> = VocabNetworkRecordItem<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.