PCGrad¶

class torchjd.aggregation.PCGrad[source]¶: Aggregator as defined in algorithm 1 of Gradient Surgery for Multi-Task Learning.

class torchjd.aggregation.PCGradWeighting[source]¶: Weighting giving the weights of PCGrad.