1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
//! Transformer models.
pub mod albert;
pub mod bert;
mod encoder;
pub use encoder::Encoder;
mod layer_output;
pub use layer_output::{HiddenLayer, LayerOutput};
pub mod roberta;
pub mod sinusoidal;
pub mod squeeze_albert;
pub mod squeeze_bert;
mod traits;