Pytorch-PCGrad:Pytorch重新实现“多任务学习的梯度手术”

PyTorch-PCGrad 该存储库提供了PyTorch 1.6.0中的重新实现代码。 设置 通过以下方式安装所需的软件包： pip install -r requirements.txt 用法 import torch import torch . nn as nn import torch . optim as optim from pcgrad import PCGrad # wrap your favorite optimizer optimizer = PCGrad ( optim . Adam ( net . parameters ())) losses = [...] # a list of per-task losses assert len ( losses ) == num_tasks optimizer . pc_backward ( losses ) # c