pub struct MetaLearningConfig {Show 19 fields
pub algorithm: MetaAlgorithm,
pub inner_lr: f64,
pub meta_lr: f64,
pub inner_steps: usize,
pub support_size: usize,
pub query_size: usize,
pub num_ways: usize,
pub num_shots: usize,
pub first_order: bool,
pub temperature: f64,
pub embedding_dim: usize,
pub normalize_embeddings: bool,
pub memory_size: usize,
pub memory_key_dim: usize,
pub memory_value_dim: usize,
pub meta_batch_size: usize,
pub task_specific_params: bool,
pub inner_l2_reg: f64,
pub grad_clip_norm: f64,
}Expand description
Configuration for meta-learning
Fields§
§algorithm: MetaAlgorithmMeta-learning algorithm to use
inner_lr: f64Inner loop learning rate (task-specific adaptation)
meta_lr: f64Meta learning rate (across-task learning)
inner_steps: usizeNumber of inner loop gradient steps
support_size: usizeNumber of support examples per task
query_size: usizeNumber of query examples per task
num_ways: usizeNumber of ways (classes) per task
num_shots: usizeNumber of shots (examples per class) per task
first_order: boolWhether to use first-order approximation
temperature: f64Temperature for softmax in prototypical networks
embedding_dim: usizeDimension of learned embeddings
normalize_embeddings: boolWhether to normalize embeddings
memory_size: usizeMemory size for memory-augmented networks
memory_key_dim: usizeMemory key dimension
memory_value_dim: usizeMemory value dimension
meta_batch_size: usizeNumber of meta-training tasks per batch
task_specific_params: boolWhether to use task-specific parameters
inner_l2_reg: f64L2 regularization for inner loop
grad_clip_norm: f64Gradient clipping threshold
Trait Implementations§
Source§impl Clone for MetaLearningConfig
impl Clone for MetaLearningConfig
Source§fn clone(&self) -> MetaLearningConfig
fn clone(&self) -> MetaLearningConfig
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 MetaLearningConfig
impl Debug for MetaLearningConfig
Source§impl Default for MetaLearningConfig
impl Default for MetaLearningConfig
Source§impl<'de> Deserialize<'de> for MetaLearningConfig
impl<'de> Deserialize<'de> for MetaLearningConfig
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
Auto Trait Implementations§
impl Freeze for MetaLearningConfig
impl RefUnwindSafe for MetaLearningConfig
impl Send for MetaLearningConfig
impl Sync for MetaLearningConfig
impl Unpin for MetaLearningConfig
impl UnsafeUnpin for MetaLearningConfig
impl UnwindSafe for MetaLearningConfig
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