pub struct ElasticAveraging { /* private fields */ }Expand description
Elastic Averaging SGD optimizer.
Implementations§
Source§impl ElasticAveraging
impl ElasticAveraging
Sourcepub fn new(
config: ElasticAveragingConfig,
worker_id: usize,
global_parameters: Arc<RwLock<Vec<Tensor>>>,
) -> Result<Self>
pub fn new( config: ElasticAveragingConfig, worker_id: usize, global_parameters: Arc<RwLock<Vec<Tensor>>>, ) -> Result<Self>
Create a new Elastic Averaging SGD optimizer.
Sourcepub fn step(&mut self, gradients: &[Tensor]) -> Result<()>
pub fn step(&mut self, gradients: &[Tensor]) -> Result<()>
Perform optimization step with elastic averaging.
Sourcepub fn get_parameters(&self) -> &[Tensor]
pub fn get_parameters(&self) -> &[Tensor]
Get current local parameters.
Auto Trait Implementations§
impl Freeze for ElasticAveraging
impl !RefUnwindSafe for ElasticAveraging
impl Send for ElasticAveraging
impl Sync for ElasticAveraging
impl Unpin for ElasticAveraging
impl !UnwindSafe for ElasticAveraging
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> 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