Expand description
Training utilities and infrastructure
This module provides comprehensive utilities for training neural networks, including advanced features like gradient accumulation, mixed precision training, distributed training, and sophisticated training loop management.
Re-exports§
pub use gradient_accumulation::*;
pub use gradient_checkpointing::*;
pub use mixed_precision::*;
pub use progress_monitor::*;
pub use quantization_aware::*;
pub use sparse_training::*;
Modules§
- gradient_
accumulation - Gradient accumulation utilities
- gradient_
checkpointing - Gradient checkpointing utilities
- mixed_
precision - Mixed precision training utilities
- progress_
monitor - Progress monitoring utilities
- quantization_
aware - Quantization-aware training utilities
- sparse_
training - Sparse training utilities
Structs§
- Training
Config - Configuration structure for training neural networks
- Training
Session - Training session for tracking training history
- Validation
Settings - Configuration for validation during training