Skip to main content

Reproducing Results for PCGrad PyTorch

PyTorch implementation for "Gradient Surgery for Multi-Task Learning" https://arxiv.org/abs/2001.06782
Created on July 6|Last edited on July 6

MultiMNIST

Notes: Runs with the substring "pcgrad-active" outperform runs with "pcgrad-disabled". "Mgda" prefixed runs are those that activate the algorithm in https://github.com/intel-isl/MultiObjectiveOptimization




5k10k15k20kStep0.40.60.811.2
5k10k15k20kStep0.80.850.90.951
Run set
5