pub struct ComputeOptimizationManager { /* private fields */ }Expand description
Compute optimization manager
Implementations§
Source§impl ComputeOptimizationManager
impl ComputeOptimizationManager
pub fn new( mixed_precision_config: AdvancedMixedPrecisionConfig, dynamic_batching_config: DynamicBatchingConfig, ) -> Self
Sourcepub fn update_step(&mut self, memory_usage: f32, performance_score: f32)
pub fn update_step(&mut self, memory_usage: f32, performance_score: f32)
Update with training step information
Sourcepub fn get_current_batch_size(&self) -> usize
pub fn get_current_batch_size(&self) -> usize
Get current batch size
Sourcepub fn get_current_precision(&self) -> &str
pub fn get_current_precision(&self) -> &str
Get current precision
Sourcepub fn generate_report(&self) -> ComputeOptimizationReport
pub fn generate_report(&self) -> ComputeOptimizationReport
Generate comprehensive optimization report
Trait Implementations§
Auto Trait Implementations§
impl Freeze for ComputeOptimizationManager
impl RefUnwindSafe for ComputeOptimizationManager
impl Send for ComputeOptimizationManager
impl Sync for ComputeOptimizationManager
impl Unpin for ComputeOptimizationManager
impl UnsafeUnpin for ComputeOptimizationManager
impl UnwindSafe for ComputeOptimizationManager
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