pub struct MixedPrecisionStats {
pub compute_dtype: PrecisionMode,
pub param_dtype: PrecisionMode,
pub current_scale: f64,
pub total_steps: usize,
pub successful_steps: usize,
pub skipped_steps: usize,
pub overflow_count: usize,
pub overflow_rate: f64,
pub success_rate: f64,
}Expand description
Statistics about mixed precision training.
Fields§
§compute_dtype: PrecisionModeCompute dtype
param_dtype: PrecisionModeParameter dtype
current_scale: f64Current loss scale
total_steps: usizeTotal training steps
successful_steps: usizeSuccessful steps (no overflow)
skipped_steps: usizeSkipped steps (due to overflow)
overflow_count: usizeTotal overflow count
overflow_rate: f64Overflow rate
success_rate: f64Success rate
Trait Implementations§
Source§impl Clone for MixedPrecisionStats
impl Clone for MixedPrecisionStats
Source§fn clone(&self) -> MixedPrecisionStats
fn clone(&self) -> MixedPrecisionStats
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for MixedPrecisionStats
impl Debug for MixedPrecisionStats
Source§impl Display for MixedPrecisionStats
impl Display for MixedPrecisionStats
Source§impl PartialEq for MixedPrecisionStats
impl PartialEq for MixedPrecisionStats
impl StructuralPartialEq for MixedPrecisionStats
Auto Trait Implementations§
impl Freeze for MixedPrecisionStats
impl RefUnwindSafe for MixedPrecisionStats
impl Send for MixedPrecisionStats
impl Sync for MixedPrecisionStats
impl Unpin for MixedPrecisionStats
impl UnwindSafe for MixedPrecisionStats
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