Trait burn_core::config::Config

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

Configuration trait.

Provided Methods§

source

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

Saves the configuration to a file.

Arguments
  • file - File to save the configuration to.
Returns

The output of the save operation.

source

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

Loads the configuration from a file.

Arguments
  • file - File to load the configuration from.
Returns

The loaded configuration.

source

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§

source§

impl Config for GradientClippingConfig

source§

impl Config for Initializer

source§

impl Config for PaddingConfig1d

source§

impl Config for PaddingConfig2d

source§

impl Config for NoamLrSchedulerConfig

source§

impl Config for MultiHeadAttentionConfig

source§

impl Config for Conv1dConfig

source§

impl Config for Conv2dConfig

source§

impl Config for ConvTranspose1dConfig

source§

impl Config for ConvTranspose2dConfig

source§

impl Config for GruConfig

source§

impl Config for BinaryCrossEntropyLossConfig

source§

impl Config for CrossEntropyLossConfig

source§

impl Config for LstmConfig

source§

impl Config for AdaptiveAvgPool1dConfig

source§

impl Config for AdaptiveAvgPool2dConfig

source§

impl Config for AvgPool1dConfig

source§

impl Config for AvgPool2dConfig

source§

impl Config for MaxPool1dConfig

source§

impl Config for MaxPool2dConfig

source§

impl Config for BatchNormConfig

source§

impl Config for DropoutConfig

source§

impl Config for EmbeddingConfig

source§

impl Config for GroupNormConfig

source§

impl Config for LayerNormConfig

source§

impl Config for LinearConfig

source§

impl Config for PositionalEncodingConfig

source§

impl Config for Unfold4dConfig

source§

impl Config for PositionWiseFeedForwardConfig

source§

impl Config for TransformerDecoderConfig

source§

impl Config for TransformerEncoderConfig

source§

impl Config for WeightDecayConfig

source§

impl Config for MomentumConfig

source§

impl Config for AdaGradConfig

source§

impl Config for AdamConfig

source§

impl Config for AdamWConfig

source§

impl Config for RMSPropConfig

source§

impl Config for SgdConfig