Struct ModelProfiler 

pub struct ModelProfiler {
    pub track_memory: bool,
    pub track_time: bool,
    pub track_activations: bool,
}

Advanced model profiler that can track memory usage and compute statistics

Fields

track_memory: bool

Whether to track memory usage

track_time: bool

Whether to track computation time

track_activations: bool

Whether to track activations

Implementations

impl ModelProfiler

pub fn new() -> Self

Create a new model profiler

pub fn with_memory_tracking(self) -> Self

Enable memory tracking

pub fn with_time_tracking(self) -> Self

Enable time tracking

pub fn with_activation_tracking(self) -> Self

Enable activation tracking

pub fn profile<M: Module>( &self, model: &M, input: &Tensor, ) -> Result<ProfileResult>

Profile a model with the given input

Trait Implementations

impl Default for ModelProfiler

fn default() -> Self

Returns the “default value” for a type.