pub enum InitializationStrategy {
Normal {
std: f32,
},
XavierUniform,
XavierNormal,
KaimingUniform,
KaimingNormal,
TruncatedNormal {
std: f32,
bounds: f32,
},
Custom(String),
}Expand description
Weight initialization strategy
Variants§
Normal
Normal distribution with mean=0, std=initializer_range
XavierUniform
Xavier/Glorot uniform initialization
XavierNormal
Xavier/Glorot normal initialization
KaimingUniform
Kaiming/He uniform initialization
KaimingNormal
Kaiming/He normal initialization
TruncatedNormal
Truncated normal initialization
Custom(String)
Custom initialization function
Trait Implementations§
Source§impl Clone for InitializationStrategy
impl Clone for InitializationStrategy
Source§fn clone(&self) -> InitializationStrategy
fn clone(&self) -> InitializationStrategy
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for InitializationStrategy
impl Debug for InitializationStrategy
Source§impl Default for InitializationStrategy
impl Default for InitializationStrategy
Source§impl<'de> Deserialize<'de> for InitializationStrategy
impl<'de> Deserialize<'de> for InitializationStrategy
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Source§impl PartialEq for InitializationStrategy
impl PartialEq for InitializationStrategy
Source§impl Serialize for InitializationStrategy
impl Serialize for InitializationStrategy
impl StructuralPartialEq for InitializationStrategy
Auto Trait Implementations§
impl Freeze for InitializationStrategy
impl RefUnwindSafe for InitializationStrategy
impl Send for InitializationStrategy
impl Sync for InitializationStrategy
impl Unpin for InitializationStrategy
impl UnsafeUnpin for InitializationStrategy
impl UnwindSafe for InitializationStrategy
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> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> ConfigSerializable for Twhere
T: Serialize + for<'de> Deserialize<'de>,
impl<T> ConfigSerializable for Twhere
T: Serialize + for<'de> Deserialize<'de>,
Source§fn save_to_file(&self, path: &Path) -> Result<(), TrustformersError>
fn save_to_file(&self, path: &Path) -> Result<(), TrustformersError>
Save to file
Source§fn load_from_file(path: &Path) -> Result<Self, TrustformersError>where
Self: Sized,
fn load_from_file(path: &Path) -> Result<Self, TrustformersError>where
Self: Sized,
Load from file
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