Trait burn::prelude::Config

pub trait Config: Serialize + DeserializeOwned {
    // Provided methods
    fn save<P>(&self, file: P) -> Result<(), Error>
       where P: AsRef<Path> { ... }
    fn load<P>(file: P) -> Result<Self, ConfigError>
       where P: AsRef<Path> { ... }
    fn load_binary(data: &[u8]) -> Result<Self, ConfigError> { ... }
}

Configuration trait.

Provided Methods

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

where P: AsRef<Path>,

Saves the configuration to a file.

Arguments

• file - File to save the configuration to.

Returns

The output of the save operation.

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

where P: AsRef<Path>,

Loads the configuration from a file.

Arguments

• file - File to load the configuration from.

Returns

The loaded configuration.

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

Loads the configuration from a binary buffer.

Arguments

• data - Binary buffer to load the configuration from.

Returns

The loaded configuration.

Object Safety

This trait is not object safe.

Implementors

impl Config for GradientClippingConfig

impl Config for Initializer

impl Config for PaddingConfig1d

impl Config for PaddingConfig2d

impl Config for PaddingConfig3d

impl Config for CosineAnnealingLrSchedulerConfig

impl Config for ExponentialLrSchedulerConfig

impl Config for LinearLrSchedulerConfig

impl Config for NoamLrSchedulerConfig

impl Config for MultiHeadAttentionConfig

impl Config for Conv1dConfig

impl Config for Conv2dConfig

impl Config for Conv3dConfig

impl Config for ConvTranspose1dConfig

impl Config for ConvTranspose2dConfig

impl Config for ConvTranspose3dConfig

impl Config for DeformConv2dConfig

impl Config for GruConfig

impl Config for Interpolate1dConfig

impl Config for Interpolate2dConfig

impl Config for BinaryCrossEntropyLossConfig

impl Config for CrossEntropyLossConfig

impl Config for HuberLossConfig

impl Config for AdaptiveAvgPool1dConfig

impl Config for AdaptiveAvgPool2dConfig

impl Config for AvgPool1dConfig

impl Config for AvgPool2dConfig

impl Config for MaxPool1dConfig

impl Config for MaxPool2dConfig

impl Config for BatchNormConfig

impl Config for BiLstmConfig

impl Config for DropoutConfig

impl Config for EmbeddingConfig

impl Config for GroupNormConfig

impl Config for HardSigmoidConfig

impl Config for InstanceNormConfig

impl Config for LayerNormConfig

impl Config for LeakyReluConfig

impl Config for LinearConfig

impl Config for LstmConfig

impl Config for PReluConfig

impl Config for PositionalEncodingConfig

impl Config for RmsNormConfig

impl Config for RotaryEncodingConfig

impl Config for SwiGluConfig

impl Config for Unfold4dConfig

impl Config for PositionWiseFeedForwardConfig

impl Config for TransformerDecoderConfig

impl Config for TransformerEncoderConfig

impl Config for WeightDecayConfig

impl Config for MomentumConfig

impl Config for AdaGradConfig

impl Config for AdamConfig

impl Config for AdamWConfig

impl Config for RmsPropConfig

impl Config for SgdConfig