Struct ModelMonitor 

pub struct ModelMonitor { /* private fields */ }

Central monitoring system for tracking model performance and resource usage

Implementations

impl ModelMonitor

pub fn new() -> Self

pub fn with_config(config: MonitoringConfig) -> Self

pub fn start_forward_pass( &mut self, batch_size: usize, sequence_length: usize, ) -> Result<MonitoringSession>

Start monitoring a forward pass

pub fn track_attention( &mut self, session: &MonitoringSession, layer_idx: usize, attention_weights: &Tensor, input_tokens: Option<&[String]>, ) -> Result<()>

Track attention weights for visualization

pub fn track_memory( &mut self, session: &MonitoringSession, checkpoint: &str, ) -> Result<MemorySnapshot>

Track memory usage at a specific point

pub fn end_session( &mut self, session: MonitoringSession, ) -> Result<MonitoringReport>

End monitoring session and collect results

pub fn set_enabled(&mut self, enabled: bool)

Enable or disable monitoring

pub fn is_enabled(&self) -> bool

Get current monitoring status

pub fn clear(&mut self) -> Result<()>

Clear all collected data

Trait Implementations

impl Clone for ModelMonitor

fn clone(&self) -> ModelMonitor

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ModelMonitor

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ModelMonitor

fn default() -> Self

Returns the “default value” for a type.