PyTorch implementation for "Gradient Surgery for Multi-Task Learning" https://arxiv.org/abs/2001.06782
Primary LanguagePythonMIT LicenseMIT