LibMTL
latest

Getting Started:

  • Introduction
  • Installation
  • Quick Start

User Guide:

  • What is Multi-Task Learning?
  • Overall Framework
  • Run a Benchmark

Developer Guide:

  • Apply to a New Dataset
  • Customize an Architecture
  • Customize a Weighting Strategy

API Reference:

  • LibMTL
  • LibMTL.loss
  • LibMTL.utils
  • LibMTL.model
  • LibMTL.config
  • LibMTL.metrics
  • LibMTL.weighting
  • LibMTL.architecture
LibMTL
  • Overview: module code

All modules for which code is available

  • LibMTL.architecture.CGC
  • LibMTL.architecture.Cross_stitch
  • LibMTL.architecture.DSelect_k
  • LibMTL.architecture.HPS
  • LibMTL.architecture.LTB
  • LibMTL.architecture.MMoE
  • LibMTL.architecture.MTAN
  • LibMTL.architecture.PLE
  • LibMTL.architecture.abstract_arch
  • LibMTL.config
  • LibMTL.loss
  • LibMTL.metrics
  • LibMTL.model.resnet
  • LibMTL.model.resnet_dilated
  • LibMTL.trainer
  • LibMTL.utils
  • LibMTL.weighting.Aligned_MTL
  • LibMTL.weighting.CAGrad
  • LibMTL.weighting.DB_MTL
  • LibMTL.weighting.DWA
  • LibMTL.weighting.EW
  • LibMTL.weighting.GLS
  • LibMTL.weighting.GradDrop
  • LibMTL.weighting.GradNorm
  • LibMTL.weighting.GradVac
  • LibMTL.weighting.IMTL
  • LibMTL.weighting.MGDA
  • LibMTL.weighting.MoCo
  • LibMTL.weighting.Nash_MTL
  • LibMTL.weighting.PCGrad
  • LibMTL.weighting.RLW
  • LibMTL.weighting.UW
  • LibMTL.weighting.abstract_weighting

© Copyright 2021, Baijiong Lin. Revision 3e603385.

Built with Sphinx using a theme provided by Read the Docs.