pub enum TaskWeightingStrategy {
Equal,
Manual(Vec<f64>),
UncertaintyWeighting,
DynamicWeighting {
adaptation_rate: f64,
min_weight: f64,
max_weight: f64,
},
GradNorm {
alpha: f64,
initial_weights: Vec<f64>,
},
}Expand description
Task weighting strategies for multi-task loss balancing
Variants§
Equal
Equal weights for all tasks
Manual(Vec<f64>)
Manual task weights
UncertaintyWeighting
Uncertainty-based weighting (homoscedastic uncertainty)
DynamicWeighting
Dynamic weight adjustment based on task difficulty
Fields
GradNorm
Gradient normalization based weighting
Trait Implementations§
Source§impl Clone for TaskWeightingStrategy
impl Clone for TaskWeightingStrategy
Source§fn clone(&self) -> TaskWeightingStrategy
fn clone(&self) -> TaskWeightingStrategy
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · 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 TaskWeightingStrategy
impl Debug for TaskWeightingStrategy
Source§impl Default for TaskWeightingStrategy
impl Default for TaskWeightingStrategy
Source§fn default() -> TaskWeightingStrategy
fn default() -> TaskWeightingStrategy
Returns the “default value” for a type. Read more
Source§impl PartialEq for TaskWeightingStrategy
impl PartialEq for TaskWeightingStrategy
impl StructuralPartialEq for TaskWeightingStrategy
Auto Trait Implementations§
impl Freeze for TaskWeightingStrategy
impl RefUnwindSafe for TaskWeightingStrategy
impl Send for TaskWeightingStrategy
impl Sync for TaskWeightingStrategy
impl Unpin for TaskWeightingStrategy
impl UnsafeUnpin for TaskWeightingStrategy
impl UnwindSafe for TaskWeightingStrategy
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> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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