pub struct MultiTaskConfig {Show 13 fields
pub sharing_type: SharingType,
pub shared_dim: usize,
pub task_specific_dims: Vec<usize>,
pub enable_dynamic_weighting: bool,
pub enable_gradient_normalization: bool,
pub enable_task_attention: bool,
pub shared_learning_rate: f64,
pub task_learning_rate: f64,
pub temperature: f64,
pub enable_curriculum: bool,
pub max_concurrent_tasks: usize,
pub enable_auxiliary_tasks: bool,
pub auxiliary_task_weight: f64,
}Expand description
Configuration for multi-task learning
Fields§
§sharing_type: SharingTypeArchitecture type for parameter sharing
Dimension of shared representation
task_specific_dims: Vec<usize>Task-specific layer dimensions
enable_dynamic_weighting: boolEnable dynamic task weighting
enable_gradient_normalization: boolEnable gradient normalization
enable_task_attention: boolEnable task attention mechanism
Learning rate for shared parameters
task_learning_rate: f64Learning rate for task-specific parameters
temperature: f64Temperature for task weighting
enable_curriculum: boolEnable curriculum learning across tasks
max_concurrent_tasks: usizeMaximum number of tasks to train simultaneously
enable_auxiliary_tasks: boolEnable auxiliary tasks for regularization
auxiliary_task_weight: f64Weight for auxiliary task losses
Trait Implementations§
Source§impl Clone for MultiTaskConfig
impl Clone for MultiTaskConfig
Source§fn clone(&self) -> MultiTaskConfig
fn clone(&self) -> MultiTaskConfig
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 MultiTaskConfig
impl Debug for MultiTaskConfig
Source§impl Default for MultiTaskConfig
impl Default for MultiTaskConfig
Source§impl<'de> Deserialize<'de> for MultiTaskConfig
impl<'de> Deserialize<'de> for MultiTaskConfig
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 MultiTaskConfig
impl RefUnwindSafe for MultiTaskConfig
impl Send for MultiTaskConfig
impl Sync for MultiTaskConfig
impl Unpin for MultiTaskConfig
impl UnsafeUnpin for MultiTaskConfig
impl UnwindSafe for MultiTaskConfig
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> 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 moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<T> PolicyExt for Twhere
T: ?Sized,
impl<T> PolicyExt for Twhere
T: ?Sized,
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.