pub struct PretrainPipeline<'a> { /* private fields */ }Expand description
Pretraining pipeline orchestrator
Implementations§
Source§impl<'a> PretrainPipeline<'a>
impl<'a> PretrainPipeline<'a>
Sourcepub fn new(engine: &'a SonaEngine) -> Self
pub fn new(engine: &'a SonaEngine) -> Self
Create new pretraining pipeline
Sourcepub fn with_config(engine: &'a SonaEngine, config: PretrainConfig) -> Self
pub fn with_config(engine: &'a SonaEngine, config: PretrainConfig) -> Self
Create with custom configuration
Sourcepub fn from_engine_stats(engine: &'a SonaEngine) -> Self
pub fn from_engine_stats(engine: &'a SonaEngine) -> Self
Generate optimal config from SONA engine stats
Sourcepub fn export_package<P: AsRef<Path>>(
&self,
output_dir: P,
) -> Result<PretrainPackage, ExportError>
pub fn export_package<P: AsRef<Path>>( &self, output_dir: P, ) -> Result<PretrainPackage, ExportError>
Export complete pretraining package
Sourcepub fn generate_dpo_script(&self) -> String
pub fn generate_dpo_script(&self) -> String
Generate DPO training script for preference learning
Auto Trait Implementations§
impl<'a> Freeze for PretrainPipeline<'a>
impl<'a> !RefUnwindSafe for PretrainPipeline<'a>
impl<'a> Send for PretrainPipeline<'a>
impl<'a> Sync for PretrainPipeline<'a>
impl<'a> Unpin for PretrainPipeline<'a>
impl<'a> !UnwindSafe for PretrainPipeline<'a>
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more