pub struct ParameterServerClient { /* private fields */ }Expand description
Parameter server client for workers
Implementations§
Source§impl ParameterServerClient
impl ParameterServerClient
Sourcepub async fn initialize_parameters(
&self,
parameters: HashMap<String, Parameter>,
) -> TorshResult<()>
pub async fn initialize_parameters( &self, parameters: HashMap<String, Parameter>, ) -> TorshResult<()>
Initialize parameters on the server
Sourcepub async fn push_gradients(
&self,
gradients: HashMap<String, Tensor>,
) -> TorshResult<u64>
pub async fn push_gradients( &self, gradients: HashMap<String, Tensor>, ) -> TorshResult<u64>
Push gradients to the parameter server
Sourcepub async fn pull_parameters(
&self,
param_names: Vec<String>,
) -> TorshResult<HashMap<String, Tensor>>
pub async fn pull_parameters( &self, param_names: Vec<String>, ) -> TorshResult<HashMap<String, Tensor>>
Pull parameters from the parameter server
Sourcepub async fn get_server_stats(&self) -> TorshResult<ParameterServerStats>
pub async fn get_server_stats(&self) -> TorshResult<ParameterServerStats>
Get server statistics
Sourcepub fn get_local_version(&self) -> u64
pub fn get_local_version(&self) -> u64
Get current local version
Auto Trait Implementations§
impl Freeze for ParameterServerClient
impl RefUnwindSafe for ParameterServerClient
impl Send for ParameterServerClient
impl Sync for ParameterServerClient
impl Unpin for ParameterServerClient
impl UnsafeUnpin for ParameterServerClient
impl UnwindSafe for ParameterServerClient
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