LibMTL.weighting.DB_MTL

class DB_MTL[source]

Bases: LibMTL.weighting.abstract_weighting.AbsWeighting

An abstract class for weighting strategies.

init_param()[source]

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

backward(losses, **kwargs)[source]
Parameters:
  • losses (list) – A list of losses of each task.

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