ConFIG support
This release adds the new ConFIG aggregator, recently proposed in ConFIG: Towards Conflict-free Training of Physics Informed Neural Networks.
To import and instantiate it:
from torchjd.aggregation import ConFIG
aggregator = ConFIG()It can then be passed to the torchjd.backward or torchjd.mtl_backward function, as usual. For more details, check out the documentation.
Thanks a lot to @qiauil (main author and maintainer of the official ConFIG repo) for his help with the integration, and to @ogencoglu for suggesting the integration of ConFIG in TorchJD!
Contributors
Contributors
ogencoglu, PierreQuinton, and 2 other contributors