TorchJD / torchjd

Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
https://torchjd.org
MIT License
151 stars 0 forks source link

Improve introduction to the base class Aggregator in doc #124

Closed ValerianRey closed 2 months ago

ValerianRey commented 2 months ago

Improve introduction to the base class Aggregator in aggregation/index.rst

ValerianRey commented 2 months ago

@PierreQuinton FYI