Struct FlowMatchingRecord 

pub struct FlowMatchingRecord<B: Backend> {
    pub cond_feature_emb: <Linear<B> as Module<B>>::Record,
    pub zero_cond_embedding1: <Param<Tensor<B, 1>> as Module<B>>::Record,
    pub estimator: <LlamaTransformer<B> as Module<B>>::Record,
    pub vq_embed: <ResidualVQ<B> as Module<B>>::Record,
    pub debug_latent_steps: EmptyRecord,
}

The record type for the module.

Fields

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

The module record associative type.

zero_cond_embedding1: <Param<Tensor<B, 1>> as Module<B>>::Record

The module record associative type.

estimator: <LlamaTransformer<B> as Module<B>>::Record

The module record associative type.

vq_embed: <ResidualVQ<B> as Module<B>>::Record

The module record associative type.

debug_latent_steps: EmptyRecord

Trait Implementations

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

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