burn_dragon_train 0.5.0

Training utilities for burn_dragon
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

#![cfg_attr(not(feature = "cli"), allow(dead_code))]

mod prelude;

pub mod artifacts;
pub mod constants;
pub mod gdpo;
pub mod metrics;
pub mod runtime;
pub mod teacher;

pub mod pipeline;

#[cfg(test)]
pub(crate) mod test_support;

#[cfg(feature = "integration_test")]
pub use gdpo::{gdpo_cpu_fallbacks, gdpo_reset_cpu_fallbacks};
#[cfg(feature = "integration_test")]
pub use metrics::{loss_trace_len, loss_trace_reset, loss_trace_take};