Enum Config 

pub enum Config {
    Attention(AttentionConfig),
    BatchNorm(BatchNormConfig),
    Bias(BiasConfig),
    Cache(CacheConfig),
    Cat(CatLayer),
    Conv2d(Conv2dConfig),
    CrossEntropy(CrossEntropyLossConfig),
    Dropout(DropoutConfig),
    Embedding(EmbeddingConfig),
    Flatten(FlattenLayer<Range<isize>>),
    KQV(KQVConfig),
    LayerNorm(LayerNormConfig),
    Linear(LinearConfig),
    MaxPool2d(MaxPool2dConfig),
    Mse,
    Relu,
    Reshape(ReshapeLayer<Reshape>),
    Rotary(RotaryEncodingConfig),
    ScaleShift(ScaleShiftConfig),
    Squeeze(SqueezeLayer),
    Stack(StackLayer),
    Sum(SumLayer),
    Tanh,
    Unsqueeze(UnsqueezeLayer),
}

enumerates config for some burn layers

Variants

Attention(AttentionConfig)

BatchNorm(BatchNormConfig)

Bias(BiasConfig)

Cache(CacheConfig)

Cat(CatLayer)

Conv2d(Conv2dConfig)

CrossEntropy(CrossEntropyLossConfig)

Dropout(DropoutConfig)

Embedding(EmbeddingConfig)

Flatten(FlattenLayer<Range<isize>>)

KQV(KQVConfig)

LayerNorm(LayerNormConfig)

Linear(LinearConfig)

MaxPool2d(MaxPool2dConfig)

Mse

Relu

Reshape(ReshapeLayer<Reshape>)

Rotary(RotaryEncodingConfig)

ScaleShift(ScaleShiftConfig)

Squeeze(SqueezeLayer)

Stack(StackLayer)

Sum(SumLayer)

Tanh

Unsqueeze(UnsqueezeLayer)

Implementations

impl Config

pub fn attention(heads: usize, mask: AttentionMask) -> Self

creates an attention config

pub fn batch_norm(countfeatures: usize, epsilon: f32, momentum: f32) -> Self

creates a batch norm config

pub fn bias(dim: usize) -> Self

creates a bias config

pub fn cache(limit: usize) -> Self

creates a cache config

pub fn dropout(chance: f32) -> Self

creates a dropout config

pub fn embedding(input: usize, output: usize) -> Self

creates a embedding config

pub fn flatten<R: RangeBounds<isize>>(dims: R) -> Self

creates a flatten config

pub fn init<B: Backend>(&self, device: &B::Device) -> Layer<B>

initializes the layer

pub fn layer_norm(dim: usize) -> Self

creates a layer norm config

pub fn linear(bias: bool, input: usize, output: usize) -> Self

creates a linear config

pub fn max_pool_2d(kernel: [usize; 2], strides: [usize; 2]) -> Self

creates a max pool 2d config

pub fn relu() -> Self

creates a relu config

pub fn reshape<R: Into<Reshape>>(args: R) -> Self

creates a reshape config

pub fn rotary(distance: usize, head: usize) -> Self

creates a rotary config

pub fn scale_shift() -> Self

creates a scale shift config

pub fn set_attention_dropout(&mut self, dropout: f32) -> bool

sets the dropout if this is an attention layer

pub fn tanh() -> Self

creates a tanh config

pub fn w_scale(self, r: f32) -> Self

scales the initializer

Trait Implementations

impl Clone for Config

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Config for Config

fn save<P>(&self, file: P) -> Result<(), Error>

where P: AsRef<Path>,

Saves the configuration to a file.

fn load<P>(file: P) -> Result<Self, ConfigError>

where P: AsRef<Path>,

Loads the configuration from a file.

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for Config

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Decompose for Config

type Decomposition = Config

the decomposed type

fn compose(decomposition: Self::Decomposition) -> Self

recreates from the decomposition

fn decompose(self) -> Self::Decomposition

owned decomposition

fn decompose_cloned(&self) -> Self::Decomposition

decomposition that copies data

impl<'de> Deserialize<'de> for Config

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Config

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl From<AttentionConfig> for Config

fn from(value: AttentionConfig) -> Self

Converts to this type from the input type.

impl From<BatchNormConfig> for Config

fn from(value: BatchNormConfig) -> Self

Converts to this type from the input type.

impl From<BiasConfig> for Config

fn from(value: BiasConfig) -> Self

Converts to this type from the input type.

impl From<CatLayer> for Config

fn from(value: CatLayer) -> Self

Converts to this type from the input type.

impl From<CrossEntropyLossConfig> for Config

fn from(value: CrossEntropyLossConfig) -> Self

Converts to this type from the input type.

impl From<DropoutConfig> for Config

fn from(value: DropoutConfig) -> Self

Converts to this type from the input type.

impl From<EmbeddingConfig> for Config

fn from(value: EmbeddingConfig) -> Self

Converts to this type from the input type.

impl From<FlattenLayer<Range<isize>>> for Config

fn from(value: FlattenLayer<Range<isize>>) -> Self

Converts to this type from the input type.

impl From<LayerNormConfig> for Config

fn from(value: LayerNormConfig) -> Self

Converts to this type from the input type.

impl From<LinearConfig> for Config

fn from(value: LinearConfig) -> Self

Converts to this type from the input type.

impl From<MaxPool2dConfig> for Config

fn from(value: MaxPool2dConfig) -> Self

Converts to this type from the input type.

impl From<MseLoss> for Config

fn from(_value: MseLoss) -> Self

Converts to this type from the input type.

impl From<Relu> for Config

fn from(_value: Relu) -> Self

Converts to this type from the input type.

impl From<ReshapeLayer<Reshape>> for Config

fn from(value: ReshapeLayer<Reshape>) -> Self

Converts to this type from the input type.

impl From<RotaryEncodingConfig> for Config

fn from(value: RotaryEncodingConfig) -> Self

Converts to this type from the input type.

impl From<SqueezeLayer> for Config

fn from(value: SqueezeLayer) -> Self

Converts to this type from the input type.

impl From<StackLayer> for Config

fn from(value: StackLayer) -> Self

Converts to this type from the input type.

impl From<SumLayer> for Config

fn from(value: SumLayer) -> Self

Converts to this type from the input type.

impl From<Tanh> for Config

fn from(_value: Tanh) -> Self

Converts to this type from the input type.

impl From<UnsqueezeLayer> for Config

fn from(value: UnsqueezeLayer) -> Self

Converts to this type from the input type.

impl Serialize for Config

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.