pub enum GradientStrategy {
AllReduce,
ReduceToRoot,
NoSync,
}Expand description
Gradient averaging strategy for distributed training.
Variants§
AllReduce
Average gradients across all workers (AllReduce).
ReduceToRoot
Reduce to rank 0 only.
NoSync
No gradient sync (for inference).
Trait Implementations§
Source§impl Clone for GradientStrategy
impl Clone for GradientStrategy
Source§fn clone(&self) -> GradientStrategy
fn clone(&self) -> GradientStrategy
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 GradientStrategy
impl Debug for GradientStrategy
Source§impl PartialEq for GradientStrategy
impl PartialEq for GradientStrategy
impl Copy for GradientStrategy
impl Eq for GradientStrategy
impl StructuralPartialEq for GradientStrategy
Auto Trait Implementations§
impl Freeze for GradientStrategy
impl RefUnwindSafe for GradientStrategy
impl Send for GradientStrategy
impl Sync for GradientStrategy
impl Unpin for GradientStrategy
impl UnsafeUnpin for GradientStrategy
impl UnwindSafe for GradientStrategy
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<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
Source§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
Source§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Compare self to
key and return true if they are equal.Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
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