Struct burn_import::pytorch::PyTorchFileRecorder

pub struct PyTorchFileRecorder<PS: PrecisionSettings> { /* private fields */ }

A recorder that that loads PyTorch files (.pt) into Burn modules.

LoadArgs can be used to remap keys or file path. See LoadArgs for more information.

Implementations

impl<PS: PrecisionSettings> PyTorchFileRecorder<PS>

pub fn new() -> Self

Constructs a new PyTorchFileRecorder.

Trait Implementations

impl<PS: Clone + PrecisionSettings> Clone for PyTorchFileRecorder<PS>

fn clone(&self) -> PyTorchFileRecorder<PS>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<PS: Debug + PrecisionSettings> Debug for PyTorchFileRecorder<PS>

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

Formats the value using the given formatter.

impl<PS: Default + PrecisionSettings> Default for PyTorchFileRecorder<PS>

fn default() -> PyTorchFileRecorder<PS>

Returns the “default value” for a type.

impl<PS: PrecisionSettings, B: Backend> Recorder<B> for PyTorchFileRecorder<PS>

type Settings = PS

Type of the settings used by the recorder.

type RecordArgs = PathBuf

Arguments used to record objects.

type RecordOutput = ()

Record output type.

type LoadArgs = LoadArgs

Arguments used to load recorded objects.

fn save_item<I: Serialize>( &self, _item: I, _file: Self::RecordArgs ) -> Result<(), RecorderError>

Saves an item.

fn load_item<I: DeserializeOwned>( &self, _file: Self::LoadArgs ) -> Result<I, RecorderError>

Loads an item.

fn load<R: Record<B>>( &self, args: Self::LoadArgs, device: &B::Device ) -> Result<R, RecorderError>

Load an item from the given arguments.

fn record<R>( &self, record: R, args: Self::RecordArgs ) -> Result<Self::RecordOutput, RecorderError>

where R: Record<B>,

Records an item.