LibMTL.weighting.UW

class UW[source]

Bases: LibMTL.weighting.abstract_weighting.AbsWeighting

Uncertainty Weights (UW).

This method is proposed in Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semantics (CVPR 2018) and implemented by us.

init_param(self)[source]

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

backward(self, losses, **kwargs)[source]
Parameters

  • losses (list) – A list of losses of each task.

  • kwargs (dict) – A dictionary of hyperparameters of weighting methods.