oxidized_transformers::architectures

Trait BuildEncoderLayer

Source 
pub trait BuildEncoderLayer: Debug {
    // Required method
    fn build_encoder_layer(
        &self,
        vb: VarBuilder<'_>,
    ) -> Result<Box<dyn EncoderLayer>, BoxedError>;
}
Expand description

Trait for building encoder layers.

Required Methods§

Source

fn build_encoder_layer( &self, vb: VarBuilder<'_>, ) -> Result<Box<dyn EncoderLayer>, BoxedError>

Build a encoder layer.

Implementors§