Struct burn_core::nn::transformer::TransformerEncoderLayerRecordItem

pub struct TransformerEncoderLayerRecordItem<B: Backend, S: PrecisionSettings> {
    pub mha: <<MultiHeadAttention<B> as Module<B>>::Record as Record<B>>::Item<S>,
    pub pwff: <<PositionWiseFeedForward<B> as Module<B>>::Record as Record<B>>::Item<S>,
    pub norm_1: <<LayerNorm<B> as Module<B>>::Record as Record<B>>::Item<S>,
    pub norm_2: <<LayerNorm<B> as Module<B>>::Record as Record<B>>::Item<S>,
    pub dropout: <<Dropout as Module<B>>::Record as Record<B>>::Item<S>,
    pub norm_first: <<bool as Module<B>>::Record as Record<B>>::Item<S>,
}

The record item type for the module.

Fields

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

Field to be serialized.

pwff: <<PositionWiseFeedForward<B> as Module<B>>::Record as Record<B>>::Item<S>

Field to be serialized.

norm_1: <<LayerNorm<B> as Module<B>>::Record as Record<B>>::Item<S>

Field to be serialized.

norm_2: <<LayerNorm<B> as Module<B>>::Record as Record<B>>::Item<S>

Field to be serialized.

dropout: <<Dropout as Module<B>>::Record as Record<B>>::Item<S>

Field to be serialized.

norm_first: <<bool as Module<B>>::Record as Record<B>>::Item<S>

Field to be serialized.

Trait Implementations

impl<'de, B: Backend, S: PrecisionSettings> Deserialize<'de> for TransformerEncoderLayerRecordItem<B, S>

where <<MultiHeadAttention<B> as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned, <<PositionWiseFeedForward<B> as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned, <<LayerNorm<B> as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned, <<Dropout as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned, <<bool as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned,

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl<B: Backend, S: PrecisionSettings> Serialize for TransformerEncoderLayerRecordItem<B, S>

where <<MultiHeadAttention<B> as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned, <<PositionWiseFeedForward<B> as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned, <<LayerNorm<B> as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned, <<Dropout as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned, <<bool as Module<B>>::Record as Record<B>>::Item<S>: Serialize + DeserializeOwned,

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.