pub struct DistributedCoordinator { /* private fields */ }Expand description
Coordinator for all distributed ML operations
Implementations§
Source§impl DistributedCoordinator
impl DistributedCoordinator
Sourcepub fn new(config: DistributedConfig) -> Self
pub fn new(config: DistributedConfig) -> Self
Create a new coordinator
Sourcepub fn initialize(&mut self) -> DistributedResult<()>
pub fn initialize(&mut self) -> DistributedResult<()>
Initialize the coordinator
Sourcepub fn is_enabled(&self) -> bool
pub fn is_enabled(&self) -> bool
Check if distributed mode is enabled
Sourcepub fn moe(&self) -> Option<&DistributedMoE>
pub fn moe(&self) -> Option<&DistributedMoE>
Get the MoE layer
Sourcepub fn moe_mut(&mut self) -> Option<&mut DistributedMoE>
pub fn moe_mut(&mut self) -> Option<&mut DistributedMoE>
Get the MoE layer mutably
Sourcepub fn averager(&self) -> Option<&DecentralizedAverager>
pub fn averager(&self) -> Option<&DecentralizedAverager>
Get the averager
Sourcepub fn averager_mut(&mut self) -> Option<&mut DecentralizedAverager>
pub fn averager_mut(&mut self) -> Option<&mut DecentralizedAverager>
Get the averager mutably
Sourcepub fn is_running(&self) -> bool
pub fn is_running(&self) -> bool
Check if coordinator is running
Trait Implementations§
Auto Trait Implementations§
impl !RefUnwindSafe for DistributedCoordinator
impl !UnwindSafe for DistributedCoordinator
impl Freeze for DistributedCoordinator
impl Send for DistributedCoordinator
impl Sync for DistributedCoordinator
impl Unpin for DistributedCoordinator
impl UnsafeUnpin for DistributedCoordinator
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
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
impl<T> ErasedDestructor for Twhere
T: 'static,
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