pub struct Calibrator { /* private fields */ }Expand description
Calibration pass over a small dataset to collect weight statistics.
In a production implementation this would hook into the model’s forward pass via observers. Here we collect statistics directly from the stored weight tensors – sufficient for symmetric per-tensor weight quantisation since we do not require input activation statistics.
Implementations§
Source§impl Calibrator
impl Calibrator
Sourcepub fn record_layer(&mut self, name: String, weights: &[f32])
pub fn record_layer(&mut self, name: String, weights: &[f32])
Record a weight tensor from one linear layer.
Sourcepub fn finish(self) -> Vec<LayerCalibration>
pub fn finish(self) -> Vec<LayerCalibration>
Consume the calibrator and produce per-layer scale factors.
Trait Implementations§
Auto Trait Implementations§
impl Freeze for Calibrator
impl RefUnwindSafe for Calibrator
impl Send for Calibrator
impl Sync for Calibrator
impl Unpin for Calibrator
impl UnsafeUnpin for Calibrator
impl UnwindSafe for Calibrator
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