Struct HalfPrecisionAdapter 

pub struct HalfPrecisionAdapter { /* private fields */ }

Adapter for mixed-precision (F32/F16) model storage.

Auto-detects conversion direction from the snapshot’s dtype:

• F32 source -> cast to F16 (typical for saving)
• F16 source -> cast to F32 (typical for loading)
• Other dtypes -> passed through unchanged

The same instance works for both with_to_adapter (save) and with_from_adapter (load).

By default, converts weights in: Linear, Embedding, Conv*, LayerNorm, GroupNorm, InstanceNorm, RmsNorm, PRelu. BatchNorm is excluded because running_var underflows in F16.

Examples

Default usage (same adapter for save and load):

let adapter = HalfPrecisionAdapter::new();
// store.with_to_adapter(adapter.clone());  // F32 -> F16 on save
// store.with_from_adapter(adapter);        // F16 -> F32 on load

Exclude a module type:

let adapter = HalfPrecisionAdapter::new()
    .without_module("LayerNorm");

Add a custom module type:

let adapter = HalfPrecisionAdapter::new()
    .with_module("CustomLayer");

Implementations

impl HalfPrecisionAdapter

pub fn new() -> Self

Create a new adapter with the default set of modules.

pub fn with_module(self, module_type: impl Into<String>) -> Self

Add a module type to convert. Accepts both short ("MyLayer") and qualified ("Struct:MyLayer") forms.

Note: short names are mapped to "Struct:Name". If you have an Enum-based module, use the qualified form "Enum:MyModule" explicitly.

pub fn without_module(self, module_type: impl Into<String>) -> Self

Remove a module type from conversion. Accepts both short and qualified forms.

Trait Implementations

impl Clone for HalfPrecisionAdapter

fn clone(&self) -> HalfPrecisionAdapter

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for HalfPrecisionAdapter

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

Formats the value using the given formatter.

impl Default for HalfPrecisionAdapter

fn default() -> Self

Returns the “default value” for a type.

impl ModuleAdapter for HalfPrecisionAdapter

fn adapt(&self, snapshot: &TensorSnapshot) -> TensorSnapshot

Adapt a tensor snapshot based on its container type and parameter name

fn clone_box(&self) -> Box<dyn ModuleAdapter>

Clone the adapter into a boxed trait object

fn get_alternative_param_name( &self, _param_name: &str, _container_type: &str, ) -> Option<String>

Get alternative parameter name to try during matching

fn chain<A>(self, next: A) -> ChainAdapter

where Self: Sized + 'static, A: ModuleAdapter + 'static,

Chain adapters together, applying self first and then next.