Struct BidiLayerPairRecord 

pub struct BidiLayerPairRecord<M>
where
    M: Module + ModuleDisplay + Module,
{
    pub straight_norm: <RmsNorm as Module>::Record,
    pub reverse_norm: <RmsNorm as Module>::Record,
    pub straight_block: <M as Module>::Record,
    pub reverse_block: <M as Module>::Record,
    pub output_merge: <OutputMerge 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

straight_norm: <RmsNorm as Module>::Record

The module record associative type.

reverse_norm: <RmsNorm as Module>::Record

The module record associative type.

straight_block: <M as Module>::Record

The module record associative type.

reverse_block: <M as Module>::Record

The module record associative type.

output_merge: <OutputMerge 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 BidiLayerPairRecord<M>

where M: Module + ModuleDisplay + Module,

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