pub struct HardwareAwareConfig {
pub target: HardwareTarget,
pub base_learning_rate: f32,
pub enable_fusion: bool,
pub memory_efficient: bool,
pub use_mixed_precision: bool,
pub gradient_compression: Option<CompressionRatio>,
pub custom_kernels: bool,
}Expand description
Hardware-aware optimizer configuration
Fields§
§target: HardwareTarget§base_learning_rate: f32§enable_fusion: bool§memory_efficient: bool§use_mixed_precision: bool§gradient_compression: Option<CompressionRatio>§custom_kernels: boolTrait Implementations§
Source§impl Clone for HardwareAwareConfig
impl Clone for HardwareAwareConfig
Source§fn clone(&self) -> HardwareAwareConfig
fn clone(&self) -> HardwareAwareConfig
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 moreAuto Trait Implementations§
impl Freeze for HardwareAwareConfig
impl RefUnwindSafe for HardwareAwareConfig
impl Send for HardwareAwareConfig
impl Sync for HardwareAwareConfig
impl Unpin for HardwareAwareConfig
impl UnwindSafe for HardwareAwareConfig
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 more