pub struct DistributedTrainer { /* private fields */ }Expand description
Distributed Trainer for handling distributed training.
Implementations§
Source§impl DistributedTrainer
impl DistributedTrainer
Sourcepub fn new(
trainer: Trainer,
config: DistributedTrainingConfig,
channel: Box<dyn DistributedCommunication>,
) -> Self
pub fn new( trainer: Trainer, config: DistributedTrainingConfig, channel: Box<dyn DistributedCommunication>, ) -> Self
Create a new distributed trainer.
Sourcepub fn train(
&mut self,
train_loader: &mut DataLoader,
num_epochs: usize,
val_loader: Option<&mut DataLoader>,
) -> CoreResult<()>
pub fn train( &mut self, train_loader: &mut DataLoader, num_epochs: usize, val_loader: Option<&mut DataLoader>, ) -> CoreResult<()>
Train the model in a distributed setting.
Auto Trait Implementations§
impl Freeze for DistributedTrainer
impl !RefUnwindSafe for DistributedTrainer
impl !Send for DistributedTrainer
impl !Sync for DistributedTrainer
impl Unpin for DistributedTrainer
impl !UnwindSafe for DistributedTrainer
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> 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