Trait ModuleAdapter 

pub trait ModuleAdapter: Send + Sync {
    // Required methods
    fn adapt(&self, snapshot: &TensorSnapshot) -> TensorSnapshot;
    fn clone_box(&self) -> Box<dyn ModuleAdapter>;
}

Trait for adapting tensor snapshots between different module formats

Required Methods

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

Trait Implementations

impl Clone for Box<dyn ModuleAdapter>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

Implementors

impl ModuleAdapter for BurnToPyTorchAdapter

impl ModuleAdapter for IdentityAdapter

impl ModuleAdapter for PyTorchToBurnAdapter