LibMTL.weighting.abstract_weighting

class AbsWeighting

Bases: torch.nn.Module

An abstract class for weighting strategies.

init_param()

Define and initialize some trainable parameters required by specific weighting methods.

property backward

Args:

losses (list): A list of losses of each task.

kwargs (dict): A dictionary of hyperparameters of weighting methods.