Struct KordModelRecord 

pub struct KordModelRecord<B: Backend> {
    pub mha: <MultiHeadAttention<B> as Module<B>>::Record,
    pub norm1: <LayerNorm<B> as Module<B>>::Record,
    pub trunk_in: <Linear<B> as Module<B>>::Record,
    pub trunk_dropout: <Dropout as Module<B>>::Record,
    pub trunk_out: <Linear<B> as Module<B>>::Record,
    pub norm2: <LayerNorm<B> as Module<B>>::Record,
    pub output: <Linear<B> as Module<B>>::Record,
    pub num_chunks: <usize as Module<B>>::Record,
    pub chunk_size: <usize as Module<B>>::Record,
}

The record type for the module.

Fields

mha: <MultiHeadAttention<B> as Module<B>>::Record

The module record associative type.

norm1: <LayerNorm<B> as Module<B>>::Record

The module record associative type.

trunk_in: <Linear<B> as Module<B>>::Record

The module record associative type.

trunk_dropout: <Dropout as Module<B>>::Record

The module record associative type.

trunk_out: <Linear<B> as Module<B>>::Record

The module record associative type.

norm2: <LayerNorm<B> as Module<B>>::Record

The module record associative type.

output: <Linear<B> as Module<B>>::Record

The module record associative type.

num_chunks: <usize as Module<B>>::Record

The module record associative type.

chunk_size: <usize as Module<B>>::Record

The module record associative type.

Trait Implementations

impl<B: Backend> Record<B> for KordModelRecord<B>

type Item<S: PrecisionSettings> = KordModelRecordItem<B, 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: &B::Device, ) -> Self

Convert the given item into a record.