pub struct CrossFrameworkConverter { /* private fields */ }Expand description
Cross-framework optimizer converter
Implementations§
Source§impl CrossFrameworkConverter
impl CrossFrameworkConverter
Sourcepub fn to_universal(
&self,
config: &dyn ConfigSource,
source_framework: Framework,
) -> Result<UniversalOptimizerConfig>
pub fn to_universal( &self, config: &dyn ConfigSource, source_framework: Framework, ) -> Result<UniversalOptimizerConfig>
Convert optimizer configuration to universal format
Sourcepub fn from_universal(
&self,
config: &UniversalOptimizerConfig,
target_framework: Framework,
) -> Result<Box<dyn ConfigTarget>>
pub fn from_universal( &self, config: &UniversalOptimizerConfig, target_framework: Framework, ) -> Result<Box<dyn ConfigTarget>>
Convert from universal format to target framework
Sourcepub fn convert(
&self,
config: &dyn ConfigSource,
source_framework: Framework,
target_framework: Framework,
) -> Result<Box<dyn ConfigTarget>>
pub fn convert( &self, config: &dyn ConfigSource, source_framework: Framework, target_framework: Framework, ) -> Result<Box<dyn ConfigTarget>>
Direct conversion between frameworks
Sourcepub fn pytorch_adam_to_tensorflow(
&self,
lr: f32,
betas: (f32, f32),
eps: f32,
weight_decay: f32,
) -> Result<TensorFlowOptimizerConfig>
pub fn pytorch_adam_to_tensorflow( &self, lr: f32, betas: (f32, f32), eps: f32, weight_decay: f32, ) -> Result<TensorFlowOptimizerConfig>
Convert PyTorch Adam to TensorFlow Adam
Sourcepub fn tensorflow_adam_to_pytorch(
&self,
lr: f32,
beta_1: f32,
beta_2: f32,
epsilon: f32,
weight_decay: f32,
) -> Result<PyTorchOptimizerConfig>
pub fn tensorflow_adam_to_pytorch( &self, lr: f32, beta_1: f32, beta_2: f32, epsilon: f32, weight_decay: f32, ) -> Result<PyTorchOptimizerConfig>
Convert TensorFlow Adam to PyTorch Adam
Sourcepub fn jax_adam_to_pytorch(
&self,
learning_rate: f32,
b1: f32,
b2: f32,
eps: f32,
) -> Result<PyTorchOptimizerConfig>
pub fn jax_adam_to_pytorch( &self, learning_rate: f32, b1: f32, b2: f32, eps: f32, ) -> Result<PyTorchOptimizerConfig>
Convert JAX Adam to PyTorch Adam
Sourcepub fn onnx_to_pytorch(
&self,
onnx_config: &ONNXConfig,
) -> Result<PyTorchOptimizerConfig>
pub fn onnx_to_pytorch( &self, onnx_config: &ONNXConfig, ) -> Result<PyTorchOptimizerConfig>
Convert ONNX optimizer to PyTorch
Sourcepub fn batch_convert(
&self,
configs: Vec<(&dyn ConfigSource, Framework)>,
target_framework: Framework,
) -> Result<Vec<Box<dyn ConfigTarget>>>
pub fn batch_convert( &self, configs: Vec<(&dyn ConfigSource, Framework)>, target_framework: Framework, ) -> Result<Vec<Box<dyn ConfigTarget>>>
Batch convert multiple optimizers
Sourcepub fn generate_conversion_report(
&self,
source: Framework,
target: Framework,
) -> String
pub fn generate_conversion_report( &self, source: Framework, target: Framework, ) -> String
Generate conversion report
Trait Implementations§
Auto Trait Implementations§
impl Freeze for CrossFrameworkConverter
impl RefUnwindSafe for CrossFrameworkConverter
impl Send for CrossFrameworkConverter
impl Sync for CrossFrameworkConverter
impl Unpin for CrossFrameworkConverter
impl UnsafeUnpin for CrossFrameworkConverter
impl UnwindSafe for CrossFrameworkConverter
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
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
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