pub struct DistributedMLConfig {
pub num_workers: usize,
pub training_mode: TrainingMode,
pub aggregation_strategy: AggregationStrategy,
pub batch_size_per_worker: usize,
pub learning_rate: f64,
pub max_epochs: usize,
pub checkpoint_interval: usize,
pub checkpoint_dir: PathBuf,
pub enable_fault_tolerance: bool,
pub health_check_interval: Duration,
pub max_gradient_staleness: usize,
}Expand description
Configuration for distributed ML training
Fields§
§num_workers: usizeNumber of worker nodes
training_mode: TrainingModeTraining mode (DataParallel or ModelParallel)
aggregation_strategy: AggregationStrategyGradient aggregation strategy
batch_size_per_worker: usizeBatch size per worker
learning_rate: f64Learning rate
max_epochs: usizeMaximum number of epochs
checkpoint_interval: usizeCheckpoint interval (in epochs)
checkpoint_dir: PathBufCheckpoint directory
enable_fault_tolerance: boolEnable fault tolerance
health_check_interval: DurationWorker health check interval
max_gradient_staleness: usizeMaximum gradient staleness (for async training)
Trait Implementations§
Source§impl Clone for DistributedMLConfig
impl Clone for DistributedMLConfig
Source§fn clone(&self) -> DistributedMLConfig
fn clone(&self) -> DistributedMLConfig
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 DistributedMLConfig
impl Debug for DistributedMLConfig
Source§impl Default for DistributedMLConfig
impl Default for DistributedMLConfig
Source§impl<'de> Deserialize<'de> for DistributedMLConfig
impl<'de> Deserialize<'de> for DistributedMLConfig
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for DistributedMLConfig
impl RefUnwindSafe for DistributedMLConfig
impl Send for DistributedMLConfig
impl Sync for DistributedMLConfig
impl Unpin for DistributedMLConfig
impl UnwindSafe for DistributedMLConfig
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> 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 moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> PolicyExt for Twhere
T: ?Sized,
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.