burn_dragon_language 0.4.0

Language modeling components for burn_dragon
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
pub mod core;
#[cfg(feature = "train")]
pub mod train;
#[cfg(all(test, feature = "train"))]
mod tests;

pub use core::{ContextStrategyConfig, GenerationConfig, ModelOverrides};
#[cfg(feature = "train")]
pub use train::{
    DatasetConfig, DatasetSourceConfig, HuggingFaceDatasetConfig, HuggingFaceRecordFormat,
    TrainingConfig, TrainingHyperparameters, load_training_config,
};