trustformers-models 0.1.1

Model implementations for TrustformeRS
Documentation
1
2
3
4
5
6
7
8
9
10
11
pub mod config;
pub mod model;
pub mod tasks;

pub use config::AlbertConfig;
pub use model::{AlbertModel, AlbertModelOutput};
pub use tasks::{
    AlbertForMaskedLM, AlbertForQuestionAnswering, AlbertForQuestionAnsweringOutput,
    AlbertForSequenceClassification, AlbertForTokenClassification, AlbertMaskedLMOutput,
    AlbertSequenceClassifierOutput, AlbertTokenClassifierOutput,
};