pub struct ZeRoStats {
pub stage: ZeRoStage,
pub world_size: usize,
pub rank: usize,
pub num_partitioned_params: usize,
pub total_param_elements: usize,
pub total_grad_elements: usize,
pub memory_savings: f32,
pub cpu_offload_enabled: bool,
}Expand description
ZeRO statistics
Fields§
§stage: ZeRoStage§world_size: usize§rank: usize§num_partitioned_params: usize§total_param_elements: usize§total_grad_elements: usize§memory_savings: f32§cpu_offload_enabled: boolTrait Implementations§
Auto Trait Implementations§
impl Freeze for ZeRoStats
impl RefUnwindSafe for ZeRoStats
impl Send for ZeRoStats
impl Sync for ZeRoStats
impl Unpin for ZeRoStats
impl UnwindSafe for ZeRoStats
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
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more