IMTL-G¶

class torchjd.aggregation.IMTLG[source]¶: Aggregator generalizing the method described in Towards Impartial Multi-task Learning. This generalization, defined formally in Jacobian Descent For Multi-Objective Optimization, supports matrices with some linearly dependant rows.

class torchjd.aggregation.IMTLGWeighting[source]¶: Weighting giving the weights of IMTLG.