scirs2_neural/training/
mixed_precision.rs

1//! Mixed precision training utilities
2
3/// Configuration for mixed precision training
4#[derive(Debug, Clone)]
5pub struct MixedPrecisionConfig {
6    /// Whether to enable mixed precision training
7    pub enabled: bool,
8    /// Loss scaling factor
9    pub loss_scale: f32,
10}
11
12impl Default for MixedPrecisionConfig {
13    fn default() -> Self {
14        Self {
15            enabled: false,
16            loss_scale: 1.0,
17        }
18    }
19}