Trait rust_bert::Config

pub trait Config<T> where
    T: Deserialize<'de>, {
    fn from_file<P: AsRef<Path>>(path: P) -> T { ... }
}

Utility to deserialize JSON config files

Provided methods

fn from_file<P: AsRef<Path>>(path: P) -> T

Loads a Config object from a JSON file. The format is expected to be aligned with the Transformers library configuration files for each model. The parsing will fail if non-optional keys expected by the model are missing.

Arguments

• path - Path to the configuration JSON file.

Example

use rust_bert::gpt2::Gpt2Config;
use rust_bert::Config;
use std::path::Path;

let config_path = Path::new("path/to/config.json");
let config = Gpt2Config::from_file(config_path);

Implementors

impl Config<AlbertConfig> for AlbertConfig

impl Config<BartConfig> for BartConfig

impl Config<BertConfig> for BertConfig

impl Config<DistilBertConfig> for DistilBertConfig

impl Config<ElectraConfig> for ElectraConfig

impl Config<Gpt2Config> for Gpt2Config

impl Config<GptNeoConfig> for GptNeoConfig

impl Config<LongformerConfig> for LongformerConfig

impl Config<MobileBertConfig> for MobileBertConfig

impl Config<ProphetNetConfig> for ProphetNetConfig

impl Config<ReformerConfig> for ReformerConfig

impl Config<T5Config> for T5Config

impl Config<XLNetConfig> for XLNetConfig