trustformers-models 0.1.1

Model implementations for TrustformeRS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

pub mod config;
pub mod model;
pub mod tasks;

#[cfg(test)]
mod tests;

pub use config::ViTConfig;
pub use model::{
    PatchEmbedding, ViTAttention, ViTEmbeddings, ViTEncoder, ViTForImageClassification, ViTLayer,
    ViTMLP, ViTModel,
};
pub use tasks::{
    multi_label_predict, predict_class, top_k_predictions, ViTForFeatureExtraction,
    ViTTaskClassifier, ViTTaskError,
};